Fix curve loading

crocs-muni · Aug 2, 2023 · 214d3ce · 214d3ce
1 parent 6e76adc
commit 214d3ce
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dissect/analysis/data_processing.py b/dissect/analysis/data_processing.py
@@ -6,7 +6,6 @@
 
 import pandas as pd
 from sklearn.neighbors import LocalOutlierFactor
-from sklearn.preprocessing import MinMaxScaler
 from sklearn.cluster import KMeans
 
 import dissect.utils.database_handler as database
@@ -45,7 +44,7 @@ def project(record: Dict[str, Any]):
         with urllib.request.urlopen(req) as f:
             curves = json.loads(f.read())["data"]
 
-    return pd.DataFrame(map(project, curves)).convert_dtypes()
+    return pd.DataFrame(map(project, curves))
 
 
 def get_trait(source: str, trait_name: str, query: Dict[str, Any] = {}, skip_failed=True):