Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
0fbc71fc
Commit
0fbc71fc
authored
Jan 07, 2019
by
Taylor Robie
Browse files
fix lint errors
parent
6726c5e0
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
official/recommendation/data_test.py
official/recommendation/data_test.py
+2
-3
No files found.
official/recommendation/data_test.py
View file @
0fbc71fc
...
...
@@ -20,9 +20,9 @@ from __future__ import print_function
from
collections
import
defaultdict
import
hashlib
import
mock
import
os
import
mock
import
numpy
as
np
import
scipy.stats
import
tensorflow
as
tf
...
...
@@ -113,8 +113,7 @@ class BaseTest(tf.test.TestCase):
for
match_mlperf
in
[
True
,
False
]:
cache_path
=
os
.
path
.
join
(
self
.
temp_data_dir
,
"test_cache.pickle"
)
data
,
valid_cache
=
data_preprocessing
.
_filter_index_sort
(
self
.
rating_file
,
cache_path
=
cache_path
,
match_mlperf
=
match_mlperf
)
self
.
rating_file
,
cache_path
=
cache_path
)
assert
len
(
data
[
rconst
.
USER_MAP
])
==
NUM_USERS
assert
len
(
data
[
rconst
.
ITEM_MAP
])
==
NUM_ITEMS
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment