Merge pull request #65 from prio-data/fix_drift_casting

force casting to float64 in drift detection
prio-data · Jun 19, 2024 · f0ff1a3 · f0ff1a3
2 parents 28d28d1 + 9b3a078
commit f0ff1a3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "viewser"
-version = "6.5.2"
+version = "6.5.3"
 description = "The Views 3 CLI tool"
 authors = ["peder2911 <[email protected]>"]
 readme = "README.md"

diff --git a/viewser/commands/queryset/drift_detection.py b/viewser/commands/queryset/drift_detection.py
@@ -109,7 +109,7 @@ class InputGate:
 
     def __init__(self, df, drift_config_dict=None):
         self.config_dict = drift_config_dict
-        self.tensor_container = objects.ViewsDataframe(df).to_numpy_time_space()
+        self.tensor_container = objects.ViewsDataframe(df, cast_to_dtype=np.float64).to_numpy_time_space()
         self.numeric_part = self.tensor_container.get_numeric_part()
         self.tensor = self.numeric_part.tensor
         self.index = self.tensor_container.index