Update df creation in helper function

JuBiotech · Jun 18, 2024 · c127211 · c127211
1 parent 8466a44
commit c127211
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bletl/__init__.py b/bletl/__init__.py
@@ -36,4 +36,4 @@
     NoMeasurementData,
 )
 
-__version__ = "1.4.1"
+__version__ = "1.4.2"
diff --git a/bletl/utils.py b/bletl/utils.py
@@ -1,4 +1,5 @@
 """Contains helper functions that do not depend on other modules within this package."""
+
 import datetime
 import enum
 import pathlib
@@ -35,7 +36,7 @@ def __to_typed_cols__(
             dfout[ncol] = None
         elif issubclass(typ, enum.Enum):
             # Enum types are kept as object-series
-            dfout[ncol] = dfin[ocol].apply(lambda x: typ(x), convert_dtype=False)
+            dfout[ncol] = pandas.Series([typ(x) for x in dfin[ocol]], name=ncol, dtype=object)
         else:
             dfout[ncol] = dfin[ocol].astype(typ)
     return dfout