update type hint of usecols

MoBagel · Oct 12, 2022 · 1b2cc9b · 1b2cc9b
1 parent a26c57b
commit 1b2cc9b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/awesome_object_store/base.py b/awesome_object_store/base.py
@@ -109,8 +109,8 @@ def fget_df(
         file: UploadFile,
         column_types: dict = {},
         date_columns: List[str] = [],
-        usecols: List = None,
-        converters: dict = None,
+        usecols: Optional[List] = None,
+        converters: Optional[dict] = None,
     ) -> Optional[pd.DataFrame]:
         try:
             file_io = StringIO(str(file.file.read(), "utf-8"))

diff --git a/awesome_object_store/gcs.py b/awesome_object_store/gcs.py
@@ -125,8 +125,8 @@ def get_df(
         name: str,
         column_types: dict = {},
         date_columns: List[str] = [],
-        usecols: List = None,
-        converters: dict = None,
+        usecols: Optional[List] = None,
+        converters: Optional[dict] = None,
     ) -> Optional[pd.DataFrame]:
         """Gets data of an object and return a dataframe."""
         try:

diff --git a/awesome_object_store/minio.py b/awesome_object_store/minio.py
@@ -113,8 +113,8 @@ def get_df(
         name: str,
         column_types: dict = {},
         date_columns: List[str] = [],
-        usecols: List = None,
-        converters: dict = None,
+        usecols: Optional[List] = None,
+        converters: Optional[dict] = None,
     ) -> Optional[pd.DataFrame]:
         """Gets data of an object and return a dataframe."""
         try: