Add type_hints argument to from_dataframe method

thombashi · Mar 29, 2020 · 8238e6a · 8238e6a
1 parent ed16cd9
commit 8238e6a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tabledata/_core.py b/tabledata/_core.py
@@ -490,7 +490,12 @@ def filter_column(
         )
 
     @staticmethod
-    def from_dataframe(dataframe, table_name: str = "", max_workers: Optional[int] = None):
+    def from_dataframe(
+        dataframe,
+        table_name: str = "",
+        type_hints: Optional[Sequence[TypeHint]] = None,
+        max_workers: Optional[int] = None,
+    ):
         """
         Initialize TableData instance from a pandas.DataFrame instance.
 
@@ -502,6 +507,7 @@ def from_dataframe(dataframe, table_name: str = "", max_workers: Optional[int] =
             table_name,
             list(dataframe.columns.values),
             dataframe.values.tolist(),
+            type_hints=type_hints,
             max_workers=max_workers,
         )