fix(indexer): is trained set on train call

jina-ai · Jun 22, 2020 · 7a648ea · 7a648ea
1 parent 6e07549
commit 7a648ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_index_train.py b/tests/test_index_train.py
@@ -4,7 +4,7 @@
 from tests import JinaTestCase
 
 
-class DummyIndexer(BaseIndexer):
+class DummyIndexerTrain(BaseIndexer):
 
     def train(self, data: 'np.ndarray', *args, **kwargs):
         pass
@@ -14,7 +14,7 @@ class MyTestCase(JinaTestCase):
 
     def test_calling_train_sets_is_trained(self):
         data = np.random.rand(1, 2)
-        i = DummyIndexer(index_filename='test2.bin')
+        i = DummyIndexerTrain(index_filename='test2.bin')
 
         # Call train method on indexer
         i.train(data)