From 5896daa56895ce3a17243847242fb38ee1078e63 Mon Sep 17 00:00:00 2001
From: Deepyaman Datta <deepyaman.datta@utexas.edu>
Date: Wed, 10 Jul 2019 12:11:47 -0400
Subject: [PATCH] Annotate types for default load and save arguments

---
 kedro/contrib/io/core.py  | 4 ++--
 kedro/io/csv_local.py     | 4 ++--
 kedro/io/csv_s3.py        | 4 ++--
 kedro/io/hdf_local.py     | 4 ++--
 kedro/io/hdf_s3.py        | 4 ++--
 kedro/io/json_local.py    | 4 ++--
 kedro/io/parquet_local.py | 4 ++--
 kedro/io/pickle_local.py  | 4 ++--
 kedro/io/pickle_s3.py     | 4 ++--
 kedro/io/sql.py           | 4 ++--
 kedro/io/text_local.py    | 4 ++--
 11 files changed, 22 insertions(+), 22 deletions(-)

diff --git a/kedro/contrib/io/core.py b/kedro/contrib/io/core.py
index ed169c4884..d4c5fb7c10 100644
--- 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
index 9dcdeb0410..74da6bfa99 100644
--- 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
index af010c1377..3e16ac85d0 100644
--- 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
index 568ba5e845..cf3edec161 100644
--- 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
index 3c2313129e..0a55b3b644 100644
--- 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
index 214066f5a2..e2bccf0a8c 100644
--- 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
index 1fd154dbf8..b85bf87079 100644
--- 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
index 9299559d43..ba01dcbc69 100644
--- 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
index cc19e82c68..99e67562f0 100644
--- 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
index e94576fb94..bcd82714dc 100644
--- 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
index b308d6b9f0..12e00e79fa 100644
--- 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,