File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -100,7 +100,7 @@ def test_load_with_overrided_params(dataset):
100100
101101 # case 1: simple inference from file system
102102 inference_pipeline = Pipeline .load (
103- logging_config .dirpath , embedder_config = EmbedderConfig (TokenizerConfig (max_length = 8 ))
103+ logging_config .dirpath , embedder_config = EmbedderConfig (tokenizer_config = TokenizerConfig (max_length = 8 ))
104104 )
105105 utterances = ["123" , "hello world" ]
106106 prediction = inference_pipeline .predict (utterances )
@@ -116,7 +116,9 @@ def test_load_with_overrided_params(dataset):
116116 pipeline_optimizer .dump ()
117117 del pipeline_optimizer
118118
119- loaded_pipe = Pipeline .load (logging_config .dirpath , embedder_config = EmbedderConfig (TokenizerConfig (max_length = 8 )))
119+ loaded_pipe = Pipeline .load (
120+ logging_config .dirpath , embedder_config = EmbedderConfig (tokenizer_config = TokenizerConfig (max_length = 8 ))
121+ )
120122 prediction_v2 = loaded_pipe .predict (utterances )
121123 assert prediction == prediction_v2
122124 assert loaded_pipe .nodes [NodeType .scoring ].module ._embedder .config .tokenizer_config .max_length == 8
You can’t perform that action at this time.
0 commit comments