fix formatting

datahub-project · May 30, 2023 · 152d21e · 152d21e
1 parent f22aedd
commit 152d21e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/athena.py b/metadata-ingestion/src/datahub/ingestion/source/sql/athena.py
@@ -60,7 +60,9 @@ class CustomAthenaRestDialect(AthenaRestDialect):
     _complex_type_pattern = re.compile(r"(<.+>)")
 
     @typing.no_type_check
-    def _get_column_type(self, type_: Union[str, Dict[str, Any]]) -> TypeEngine:  # noqa: C901
+    def _get_column_type(self,
+        type_: Union[str, Dict[str, Any]]
+    ) -> TypeEngine:  # noqa: C901
         """Derives the data type of the Athena column.
 
         This method is overwritten to extend the behavior of PyAthena.

diff --git a/metadata-ingestion/src/datahub/utilities/sqlalchemy_type_converter.py b/metadata-ingestion/src/datahub/utilities/sqlalchemy_type_converter.py
@@ -119,7 +119,7 @@ def get_avro_for_sqlalchemy_column(
         column_name: str,
         column_type: types.TypeEngine,
         nullable: bool,
-    ) -> Union[object,  Dict[str, object]]:
+    ) -> Union[object, Dict[str, object]]:
         """Returns the AVRO schema representation of a SQLalchemy column."""
         if isinstance(column_type, cls._COMPLEX_TYPES):
             return {