Fix tests/deepsparse/pipelines/test_bucketing.py

neuralmagic · Apr 29, 2024 · ccaec52 · ccaec52
1 parent 164252f
commit ccaec52
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/deepsparse/transformers/pipelines/mnli_text_classification.py b/src/deepsparse/transformers/pipelines/mnli_text_classification.py
@@ -90,7 +90,8 @@ class MnliTextClassificationInput(ZeroShotTextClassificationInputBase):
     labels: Optional[Union[List[str], str]] = Field(
         description="The set of possible class labels to classify each "
         "sequence into. Can be a single label, a string of comma-separated "
-        "labels, or a list of labels."
+        "labels, or a list of labels.",
+        default=None,
     )
     hypothesis_template: Optional[str] = Field(
         description="A formattable template for wrapping around the provided "