fix(ingest): lint fix a few files

datahub-project · Jan 31, 2022 · b34ef01 · b34ef01
1 parent 646dd6d
commit b34ef01
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/state/checkpoint.py b/metadata-ingestion/src/datahub/ingestion/source/state/checkpoint.py
@@ -33,7 +33,7 @@ def to_bytes(
         compressor: Callable[[bytes], bytes] = functools.partial(
             bz2.compress, compresslevel=9
         ),
-        max_allowed_state_size: int = 2 ** 22,  # 4MB
+        max_allowed_state_size: int = 2**22,  # 4MB
     ) -> bytes:
         """
         NOTE: Binary compression cannot be turned on yet as the current MCPs encode the GeneralizedAspect

diff --git a/metadata-ingestion/src/datahub/ingestion/source/state/stateful_ingestion_base.py b/metadata-ingestion/src/datahub/ingestion/source/state/stateful_ingestion_base.py
@@ -30,7 +30,7 @@ class StatefulIngestionConfig(ConfigModel):
     """
 
     enabled: bool = False
-    max_checkpoint_state_size: int = 2 ** 24  # 16MB
+    max_checkpoint_state_size: int = 2**24  # 16MB
     state_provider: Optional[DynamicTypedConfig] = DynamicTypedConfig(
         type="datahub", config=DatahubIngestionStateProviderConfig()
     )