Bug fix: numpy float interpreted as VECTOR

comet-ml · Dec 6, 2022 · 6c7ff95 · 6c7ff95
1 parent 3c92182
commit 6c7ff95
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/backend/kangas/datatypes/utils.py b/backend/kangas/datatypes/utils.py
@@ -152,17 +152,17 @@ def pytype_to_dgtype(item):
         else:
             return "JSON"
 
-    if hasattr(item, "tolist"):
-        ## numpy arrays
-        return "VECTOR"
-
     if hasattr(item, "item") and callable(item.item):
         ## numpy types
         try:
             item = item.item()
         except Exception:
             pass
 
+    if hasattr(item, "tolist") and not isinstance(item, numbers.Number):
+        ## numpy arrays
+        return "VECTOR"
+
     for ctype in DATAGRID_TYPES:
         if isinstance(item, tuple(DATAGRID_TYPES[ctype]["types"])):
             return ctype