Change pandas_list enforce_type df construction (#72)

* Change pandas_list enforce_type df construction * Change pandas_list enforce_type df construction
fugue-project · Oct 20, 2021 · ece1754 · ece1754
1 parent 7b9fec9
commit ece1754
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -20,6 +20,10 @@ pip install triad
 
 ## Release History
 
+### 0.5.5
+
+* Change pandas_list enforce_type df construction
+
 ### 0.5.4
 
 * Make `FileSystem` work for windows

diff --git a/triad/utils/pandas_like.py b/triad/utils/pandas_like.py
@@ -143,6 +143,7 @@ def enforce_type(  # noqa: C901
             return df
         if not null_safe:
             return df.astype(dtype=to_pandas_dtype(schema))
+        data: Dict[str, Any] = {}
         for v in schema:
             s = df[v.name]
             if pa.types.is_string(v.type):
@@ -163,8 +164,8 @@ def enforce_type(  # noqa: C901
                 s = s.fillna(0).astype(v.type.to_pandas_dtype()).mask(ns, None)
             elif not pa.types.is_struct(v.type) and not pa.types.is_list(v.type):
                 s = s.astype(v.type.to_pandas_dtype())
-            df[v.name] = s
-        return df
+            data[v.name] = s
+        return pd.DataFrame(data)
 
     def safe_groupby_apply(
         self,

diff --git a/triad_version/__init__.py b/triad_version/__init__.py
@@ -1 +1 @@
-__version__ = "0.5.4"
+__version__ = "0.5.5"