fix the model being None probelm

Anindyadeep · Nov 28, 2023 · d35ac7a · d35ac7a
1 parent 4bf3ae9
commit d35ac7a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepeval/models/detoxify_model.py b/deepeval/models/detoxify_model.py
@@ -4,7 +4,8 @@
 
 class DetoxifyModel(DeepEvalBaseModel):
     def __init__(self, model_name: str | None = None, *args, **kwargs):
-        assert model_name in ["original", "unbiased", "multilingual"], "Invalid model. Available variants: original, unbiased, multilingual"
+        if model_name is not None:
+            assert model_name in ["original", "unbiased", "multilingual"], "Invalid model. Available variants: original, unbiased, multilingual"
         model_name = 'original' if model_name is None else model_name
         super().__init__(model_name, *args, **kwargs)