run black

Author: dongwonmoon

data_scribe/components/db_connectors/duckdb_connector.py

@@ -144,7 +144,8 @@ def get_columns(self, table_name: str) -> List[Dict[str, str]]:
 
         except Exception as e:
             logger.error(
-                f"Failed to fetch columns for '{table_name}': {e}", exc_info=True
+                f"Failed to fetch columns for '{table_name}': {e}",
+                exc_info=True,
             )
             raise ConnectorError(
                 f"Failed to fetch columns for '{table_name}': {e}"
@@ -229,7 +230,9 @@ def get_column_profile(
             }
             return stats
         except Exception as e:
-            logger.warning(f"Failed to profile '{table_name}.{column_name}': {e}")
+            logger.warning(
+                f"Failed to profile '{table_name}.{column_name}': {e}"
+            )
             return {
                 "null_ratio": "N/A",
                 "distinct_count": "N/A",

data_scribe/components/db_connectors/sql_base_connector.py

@@ -254,7 +254,9 @@ def get_column_profile(
                 "distinct_count": distinct_count,
                 "is_unique": is_unique,
             }
-            logger.info(f"  - Profile for '{table_name}.{column_name}': {stats}")
+            logger.info(
+                f"  - Profile for '{table_name}.{column_name}': {stats}"
+            )
             return stats
 
         except Exception as e:

data_scribe/components/db_connectors/sqlite_connector.py

@@ -182,7 +182,9 @@ def get_foreign_keys(self) -> List[Dict[str, str]]:
                         }
                     )
             except sqlite3.Error as e:
-                logger.warning(f"Failed to get FKs for table '{table_name}': {e}")
+                logger.warning(
+                    f"Failed to get FKs for table '{table_name}': {e}"
+                )
 
         logger.info(f"Found {len(foreign_keys)} foreign key relationships.")
         return foreign_keys
@@ -249,7 +251,9 @@ def get_column_profile(
                 "distinct_count": distinct_count,
                 "is_unique": is_unique,
             }
-            logger.info(f"  - Profile for '{table_name}.{column_name}': {stats}")
+            logger.info(
+                f"  - Profile for '{table_name}.{column_name}': {stats}"
+            )
             return stats
         except sqlite3.Error as e:
             logger.warning(

data_scribe/components/writers/dbt_yaml_writer.py

@@ -318,7 +318,9 @@ def _create_new_model_stub(
 
         else:
             # File does not exist, create new
-            logger.info(f"   -> Creating new file for stub: '{target_yaml_path}'")
+            logger.info(
+                f"   -> Creating new file for stub: '{target_yaml_path}'"
+            )
             new_yaml_data = CommentedMap()
             new_yaml_data["version"] = 2
             new_yaml_data["models"] = [new_model_stub]

data_scribe/components/writers/notion_writer.py

@@ -293,7 +293,7 @@ def _create_column_table(
                     self._text_cell("Data Type"),
                     self._text_cell("AI-Generated Description"),
                 ]
-            }
+            },
         }
 
         rows = [header]
@@ -313,7 +313,7 @@ def _create_column_table(
                         self._text_cell(col.get("type")),
                         self._text_cell(desc),
                     ],
-                }
+                },
             }
             rows.append(row)