feat(ingest): Couchbase source lint fix update

datahub-project · Jan 23, 2025 · 2cc5ba5 · 2cc5ba5
1 parent d45700e
commit 2cc5ba5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/couchbase/couchbase_kv_schema.py b/metadata-ingestion/src/datahub/ingestion/source/couchbase/couchbase_kv_schema.py
@@ -123,7 +123,7 @@ def discard(_tuple: Tuple[str, ...], element: str) -> Tuple[str, ...]:
 
 
 def construct_schema(
-    collection: Dict[str, Any]
+    collection: Dict[str, Any],
 ) -> Dict[Tuple[str, ...], SchemaDescription]:
     """
     Construct JSON schema.

diff --git a/metadata-ingestion/src/datahub/ingestion/source/couchbase/couchbase_main.py b/metadata-ingestion/src/datahub/ingestion/source/couchbase/couchbase_main.py
@@ -352,9 +352,9 @@ def construct_schema_metadata(
             )
             # Add this information to the custom properties so user can know they are looking at downsampled schema
             dataset_properties.customProperties["schema.downsampled"] = "True"
-            dataset_properties.customProperties[
-                "schema.totalFields"
-            ] = f"{collection_schema_size}"
+            dataset_properties.customProperties["schema.totalFields"] = (
+                f"{collection_schema_size}"
+            )
 
         logger.debug(f"Size of collection fields = {len(collection_fields)}")
         # append each schema field (sort so output is consistent)