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
ea36125a
Commit
ea36125a
authored
Jan 07, 2019
by
Taylor Robie
Browse files
fix test now that cache construction no longer uses match_mlperf
parent
fefe47ee
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
8 deletions
+6
-8
official/recommendation/data_test.py
official/recommendation/data_test.py
+6
-8
No files found.
official/recommendation/data_test.py
View file @
ea36125a
...
...
@@ -110,14 +110,12 @@ class BaseTest(tf.test.TestCase):
# For the most part the necessary checks are performed within
# _filter_index_sort()
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
)
assert
len
(
data
[
rconst
.
USER_MAP
])
==
NUM_USERS
assert
len
(
data
[
rconst
.
ITEM_MAP
])
==
NUM_ITEMS
assert
not
valid_cache
def
drain_dataset
(
self
,
dataset
,
g
):
# type: (tf.data.Dataset, tf.Graph) -> list
...
...
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