fix(flow): fix flow unit test

gnes-ai · Oct 11, 2019 · a3da058 · a3da058
1 parent 9095bfa
commit a3da058
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/tests/test_gnes_flow.py b/tests/test_gnes_flow.py
@@ -144,7 +144,7 @@ def _test_query_flow(self):
         with flow.build(backend='process') as f, open(self.test_file, encoding='utf8') as fp:
             f.query(bytes_gen=[v.encode() for v in fp][:10])
 
-    @unittest.SkipTest
+    # @unittest.SkipTest
     def test_index_query_flow(self):
         self._test_index_flow()
         print('indexing finished')

diff --git a/tests/yaml/flow-transformer.yml b/tests/yaml/flow-transformer.yml
@@ -1,13 +1,6 @@
 !PipelineEncoder
 components:
-  - !PyTorchTransformers
-    parameters:
-      model_dir: $TORCH_TRANSFORMERS_MODEL
-      model_name: bert-base-uncased
-  - !PoolingEncoder
-    parameters:
-      pooling_strategy: REDUCE_MEAN
-      backend: torch
+  - !CharEmbeddingEncoder {}
 gnes_config:
   name: my_transformer  # a customized name
   is_trained: true  # indicate the model has been trained