Skip to content

Commit 2abf1b2

Browse files
committed
fix tests
1 parent 068a714 commit 2abf1b2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/pipeline/test_inference.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)