hopeit.dataframes 0.25.0b5__tar.gz → 0.25.0b7__tar.gz

{hopeit_dataframes-0.25.0b5 → hopeit_dataframes-0.25.0b7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hopeit.dataframes
-Version: 0.25.0b5
+Version: 0.25.0b7
 Summary: Hopeit Engine Dataframes Toolkit
 Home-page: https://github.com/hopeit-git/hopeit.engine
 Author: Leo Smerling and Pablo Canto
@@ -26,7 +26,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Framework :: AsyncIO
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: hopeit.engine[fs-storage]==0.25.0b5
+Requires-Dist: hopeit.engine[fs-storage]==0.25.0b7
 Requires-Dist: pandas
 Requires-Dist: numpy
 Provides-Extra: pyarrow

{hopeit_dataframes-0.25.0b5 → hopeit_dataframes-0.25.0b7}/src/hopeit/dataframes/dataframe.py

@@ -3,6 +3,7 @@ DataFrames type abstractions.
 """
 import dataclasses
 from datetime import date, datetime, timezone
+from functools import partial
 from typing import Any, Callable, Dict, Generic, Iterator, List, Type, TypeVar
 
 import numpy as np
@@ -28,6 +29,22 @@ class DataFrameMetadata():
     fields: Dict[str, FieldInfo]
 
 
+# Functions to do type coercion
+def _series_to_int(x: pd.Series) -> pd.Series:
+    return x.astype(np.int64)
+
+
+def _series_to_float(x: pd.Series) -> pd.Series:
+    return x.astype(np.float64)
+
+
+def _series_to_str(x: pd.Series) -> pd.Series:
+    return x.astype(str)
+
+
+_series_to_utc_datetime = partial(pd.to_datetime, utc=True)
+
+
 class DataFrameMixin(Generic[DataFrameT, DataObject]):
     """
     MixIn class to add functionality for DataFrames dataobjects
@@ -36,11 +53,11 @@ class DataFrameMixin(Generic[DataFrameT, DataObject]):
     """
 
     DATATYPE_MAPPING = {
-        int: lambda x: x.astype(np.int64),
-        float: lambda x: x.astype(np.float64),
-        str: lambda x: x.astype(object),
+        int: _series_to_int,
+        float: _series_to_float,
+        str: _series_to_str,
         date: pd.to_datetime,
-        datetime: pd.to_datetime,
+        datetime: _series_to_utc_datetime,
     }
 
     def __init__(self) -> None:

{hopeit_dataframes-0.25.0b5 → hopeit_dataframes-0.25.0b7}/src/hopeit.dataframes.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hopeit.dataframes
-Version: 0.25.0b5
+Version: 0.25.0b7
 Summary: Hopeit Engine Dataframes Toolkit
 Home-page: https://github.com/hopeit-git/hopeit.engine
 Author: Leo Smerling and Pablo Canto
@@ -26,7 +26,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Framework :: AsyncIO
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: hopeit.engine[fs-storage]==0.25.0b5
+Requires-Dist: hopeit.engine[fs-storage]==0.25.0b7
 Requires-Dist: pandas
 Requires-Dist: numpy
 Provides-Extra: pyarrow

hopeit_dataframes-0.25.0b7/src/hopeit.dataframes.egg-info/requires.txt

@@ -0,0 +1,6 @@
+hopeit.engine[fs-storage]==0.25.0b7
+pandas
+numpy
+
+[pyarrow]
+pyarrow

hopeit_dataframes-0.25.0b5/src/hopeit.dataframes.egg-info/requires.txt

@@ -1,6 +0,0 @@
-hopeit.engine[fs-storage]==0.25.0b5
-pandas
-numpy
-
-[pyarrow]
-pyarrow