Commit ea36125a authored by Taylor Robie's avatar Taylor Robie
Browse files

fix test now that cache construction no longer uses match_mlperf

parent fefe47ee
...@@ -110,14 +110,12 @@ class BaseTest(tf.test.TestCase): ...@@ -110,14 +110,12 @@ class BaseTest(tf.test.TestCase):
# For the most part the necessary checks are performed within # For the most part the necessary checks are performed within
# _filter_index_sort() # _filter_index_sort()
for match_mlperf in [True, False]: cache_path = os.path.join(self.temp_data_dir, "test_cache.pickle")
cache_path = os.path.join(self.temp_data_dir, "test_cache.pickle") data, valid_cache = data_preprocessing._filter_index_sort(
data, valid_cache = data_preprocessing._filter_index_sort( self.rating_file, cache_path=cache_path)
self.rating_file, cache_path=cache_path)
assert len(data[rconst.USER_MAP]) == NUM_USERS
assert len(data[rconst.USER_MAP]) == NUM_USERS assert len(data[rconst.ITEM_MAP]) == NUM_ITEMS
assert len(data[rconst.ITEM_MAP]) == NUM_ITEMS
assert not valid_cache
def drain_dataset(self, dataset, g): def drain_dataset(self, dataset, g):
# type: (tf.data.Dataset, tf.Graph) -> list # type: (tf.data.Dataset, tf.Graph) -> list
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment