diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/athena.py b/metadata-ingestion/src/datahub/ingestion/source/sql/athena.py
index 7db2a3af176085..03ce79d9d6b232 100644
--- 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
index b76e545f0b799f..74c1b657973b1e 100644
--- 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 {