Annotate types for default load and save arguments

kedro-org · Jul 10, 2019 · 5896daa · 5896daa
1 parent 2a575d6
commit 5896daa
Show file tree

Hide file tree

Showing 11 changed files with 22 additions and 22 deletions.
diff --git a/kedro/contrib/io/core.py b/kedro/contrib/io/core.py
@@ -36,8 +36,8 @@
 class DefaultArgumentsMixIn:
     """Mixin class that helps handle default load and save arguments."""
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {}
 
     def __init__(
         self,

diff --git a/kedro/io/csv_local.py b/kedro/io/csv_local.py
@@ -62,8 +62,8 @@ class CSVLocalDataSet(AbstractVersionedDataSet):
 
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {"index": False}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {"index": False}
 
     def __init__(
         self,

diff --git a/kedro/io/csv_s3.py b/kedro/io/csv_s3.py
@@ -63,8 +63,8 @@ class CSVS3DataSet(AbstractVersionedDataSet):
         >>> assert data.equals(reloaded)
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {"index": False}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {"index": False}
 
     # pylint: disable=too-many-arguments
     def __init__(

diff --git a/kedro/io/hdf_local.py b/kedro/io/hdf_local.py
@@ -64,8 +64,8 @@ class HDFLocalDataSet(AbstractVersionedDataSet):
 
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {}
 
     # pylint: disable=too-many-arguments
     def __init__(

diff --git a/kedro/io/hdf_s3.py b/kedro/io/hdf_s3.py
@@ -69,8 +69,8 @@ class HDFS3DataSet(AbstractVersionedDataSet):
 
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {}
 
     # pylint: disable=too-many-arguments
     def __init__(

diff --git a/kedro/io/json_local.py b/kedro/io/json_local.py
@@ -59,8 +59,8 @@ class JSONLocalDataSet(AbstractVersionedDataSet):
 
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {"indent": 4}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {"indent": 4}
 
     def __init__(
         self,

diff --git a/kedro/io/parquet_local.py b/kedro/io/parquet_local.py
@@ -62,8 +62,8 @@ class ParquetLocalDataSet(AbstractVersionedDataSet):
         >>> assert data.equals(loaded_data)
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {"compression": None}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {"compression": None}
 
     # pylint: disable=too-many-arguments
     def __init__(

diff --git a/kedro/io/pickle_local.py b/kedro/io/pickle_local.py
@@ -68,8 +68,8 @@ class PickleLocalDataSet(AbstractVersionedDataSet):
         >>> reloaded = data_set.load()
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {}
 
     BACKENDS = {"pickle": pickle, "joblib": joblib}
 

diff --git a/kedro/io/pickle_s3.py b/kedro/io/pickle_s3.py
@@ -65,8 +65,8 @@ class PickleS3DataSet(AbstractVersionedDataSet):
             >>> reloaded = data_set.load()
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {}
 
     # pylint: disable=too-many-arguments
     def __init__(

diff --git a/kedro/io/sql.py b/kedro/io/sql.py
@@ -140,8 +140,8 @@ class SQLTableDataSet(AbstractDataSet):
 
     """
 
-    DEFAULT_LOAD_ARGS = {}
-    DEFAULT_SAVE_ARGS = {"index": False}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {"index": False}
 
     def _describe(self) -> Dict[str, Any]:
         load_args = self._load_args.copy()

diff --git a/kedro/io/text_local.py b/kedro/io/text_local.py
@@ -51,8 +51,8 @@ class TextLocalDataSet(AbstractVersionedDataSet):
         >>> reloaded = data_set.load()
     """
 
-    DEFAULT_LOAD_ARGS = {"mode": "r"}
-    DEFAULT_SAVE_ARGS = {"mode": "w"}
+    DEFAULT_LOAD_ARGS: Dict[str, Any] = {"mode": "r"}
+    DEFAULT_SAVE_ARGS: Dict[str, Any] = {"mode": "w"}
 
     def __init__(
         self,