return nan for float (#493)

Arize-ai · Apr 4, 2023 · 8183737 · 8183737
1 parent 756108e
commit 8183737
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/phoenix/metrics/metrics.py b/src/phoenix/metrics/metrics.py
@@ -89,9 +89,11 @@ def calc(self, dataframe: pd.DataFrame) -> float:
 
 @dataclass
 class Cardinality(UnaryOperator, BaseMetric):
-    def calc(self, dataframe: pd.DataFrame) -> int:
+    def calc(self, dataframe: pd.DataFrame) -> float:
         data = self.get_operand_column(dataframe)
-        return data.nunique()
+        if data.dtype.kind == "f":
+            return float("nan")
+        return cast(float, data.nunique())
 
 
 @dataclass