Skip to content

GitLab

Commit 0fbc71fc authored by Taylor Robie's avatar Taylor Robie
Browse files

fix lint errors

parent 6726c5e0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
official/recommendation/data_test.py
View file @ 0fbc71fc
...@@ -20,9 +20,9 @@ from __future__ import print_function ...@@ -20,9 +20,9 @@ from __future__ import print_function
from collections import defaultdict from collections import defaultdict
import hashlib import hashlib
import mock
import os import os
import mock
import numpy as np import numpy as np
import scipy.stats import scipy.stats
import tensorflow as tf import tensorflow as tf
...@@ -113,8 +113,7 @@ class BaseTest(tf.test.TestCase): ...@@ -113,8 +113,7 @@ class BaseTest(tf.test.TestCase):
for match_mlperf in [True, False]: 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)
match_mlperf=match_mlperf)
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
......
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