Skip to content
Permalink
Browse files

fix(flow): fix flow unit test

  • Loading branch information...
hanxiao committed Oct 11, 2019
1 parent 9095bfa commit a3da05829c2756c42f45a41572a9a0f2217d9d6a
Showing with 2 additions and 9 deletions.
  1. +1 −1 tests/test_gnes_flow.py
  2. +1 −8 tests/yaml/flow-transformer.yml
@@ -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')
@@ -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

0 comments on commit a3da058

Please sign in to comment.
You can’t perform that action at this time.