updated param dict

pytorch · Oct 6, 2021 · 3068a49 · 3068a49
1 parent 523c243
commit 3068a49
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/test/pytest/test_handler.py b/test/pytest/test_handler.py
@@ -228,7 +228,15 @@ def test_kfserving_mnist_model_register_and_inference_on_valid_model_explain():
 def test_huggingface_bert_batch_inference():
     batch_size = 4
     batch_delay = 10000 # 10 seconds
-    test_utils.register_model_with_params(model_name='BERTSeqClassification', url='https://bert-mar-file.s3.us-west-2.amazonaws.com/BERTSeqClassification.mar', batch_size=batch_size, batch_delay=batch_delay)
+    params = (
+        ('model_name', 'BERTSeqClassification'),
+        ('url', 'https://bert-mar-file.s3.us-west-2.amazonaws.com/BERTSeqClassification.mar'),
+        ('initial_workers', '1'),
+        ('synchronous', 'true'),
+        ('batch_size', '4'),
+        ('batch_delay', '10000')
+    )
+    test_utils.register_model_with_params(params)
     input_text = "../../examples/Huggingface_Transformers/Seq_classification_artifacts/sample_text.txt"
     files = {
     'data': (input_text,