changed JSONType to be nonrecursive and removed cast on key

capitalone · Oct 18, 2022 · 8de0cef · 8de0cef
1 parent bb9807f
commit 8de0cef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dataprofiler/_typing.py b/dataprofiler/_typing.py
@@ -5,4 +5,4 @@
 import pandas as pd
 
 DataArray = Union[pd.DataFrame, pd.Series, np.ndarray]
-JSONType = Union[str, int, float, bool, None, List["JSONType"], Dict[str, "JSONType"]]
+JSONType = Union[str, int, float, bool, None, List, Dict]
diff --git a/dataprofiler/data_readers/data_utils.py b/dataprofiler/data_readers/data_utils.py
@@ -105,7 +105,7 @@ def unicode_to_str(data: JSONType, ignore_dicts: bool = False) -> JSONType:
     # if data is a dictionary
     if isinstance(data, dict) and not ignore_dicts:
         return {
-            cast(str, unicode_to_str(key, ignore_dicts=True)): unicode_to_str(
+            unicode_to_str(key, ignore_dicts=True): unicode_to_str(
                 value, ignore_dicts=True
             )
             for key, value in data.items()