SetFit: When logging probs, log them not as logits (#612)

This is a proposed fixed that might resolve quirks around the DEP reported for setfit
rungalileo · May 18, 2023 · 0945629 · 0945629
1 parent 40b908d
commit 0945629
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dataquality/__init__.py b/dataquality/__init__.py
@@ -30,7 +30,7 @@
         dataquality.get_insights()
 """
 
-__version__ = "v0.8.42"
+__version__ = "v0.8.43"
 
 import sys
 from typing import Any, List, Optional

diff --git a/dataquality/integrations/setfit.py b/dataquality/integrations/setfit.py
@@ -246,7 +246,7 @@ def __call__(self, *args: Any, **kwds: Any) -> Any:
                 # 🔭🌕 Galileo logging
                 dq.log_model_outputs(
                     ids=batch["id"],
-                    logits=dq_store["output"],
+                    probs=dq_store["output"],
                     embs=dq_store["input_args"][0],
                     split=split,
                     epoch=0,
@@ -362,7 +362,7 @@ def dq_evaluate(
             # 🔭🌕 Galileo logging
             dq.log_model_outputs(
                 ids=batch[id_col],
-                logits=dq_store["output"],
+                probs=dq_store["output"],
                 embs=dq_store["input_args"][0],
                 split=split,
                 epoch=0,