fixed unit tests

SeldonIO · Dec 6, 2023 · e7b3123 · e7b3123
1 parent 2b8cf1b
commit e7b3123
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mlserver/types/dataplane.py b/mlserver/types/dataplane.py
@@ -28,7 +28,7 @@ class Config:
 
     content_type: Optional[str] = None
     headers: Optional[Dict[str, Any]] = None
-    extra: Optional[dict] = {}
+    extra: Optional[dict] = None
 
 
 class TensorData(BaseModel):

diff --git a/runtimes/huggingface/tests/test_common.py b/runtimes/huggingface/tests/test_common.py
@@ -1,3 +1,4 @@
+from symbol import comparison
 from unittest.mock import MagicMock, patch
 import json
 import pytest
@@ -269,4 +270,5 @@ async def test_pipeline_uses_inference_kwargs(
 
     result2 = await runtime.predict(payload2)
     result2 = json.loads(result2.outputs[0].data[0])["generated_text"]
-    assert (len(result1) > len(result2)) == expected
+    comparison = len(result1) > len(result2)
+    assert comparison == expected