fix python 3.9 typing error from mistral

wandb · Aug 14, 2024 · c9c48ae · c9c48ae
1 parent 1313970
commit c9c48ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/weave/trace/op.py b/weave/trace/op.py
@@ -46,7 +46,8 @@
     ANTHROPIC_NOT_GIVEN = None
 
 try:
-    from mistralai.types.basemodel import UNSET as MISTRAL_NOT_GIVEN
+    from mistralai.types.basemodel import UNSET 
+    MISTRAL_NOT_GIVEN = UNSET
 except ImportError:
     MISTRAL_NOT_GIVEN = None