🎨 Fix

Signed-off-by: zethson <[email protected]>
laminlabs · Aug 23, 2024 · d9726aa · d9726aa
1 parent 9ebdaaf
commit d9726aa
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lamindb/_can_validate.py b/lamindb/_can_validate.py
@@ -405,9 +405,10 @@ def _return(result: list, mapper: dict):
             logger.warning(warn_msg)
 
         mapper.update(std_names_bt_mapper)
-        result = (
-            pd.Series(std_names_db).cat.rename_categories(std_names_bt_mapper).tolist()
-        )
+        if pd.api.types.is_categorical_dtype(std_names_db):
+            result = std_names_db.cat.rename_categories(std_names_bt_mapper).tolist()
+        else:
+            result = pd.Series(std_names_db).replace(std_names_bt_mapper).tolist()
         return _return(result=result, mapper=mapper)
 
     else: