format and unused method

Author: dlawin

data_diff/dbt.py

@@ -7,10 +7,15 @@
 import rich
 from rich.prompt import Prompt
 
-from data_diff.errors import DataDiffCustomSchemaNoConfigError, DataDiffDbtProjectVarsNotFoundError, DataDiffNoAPIKeyError, DataDiffNoDatasourceIdError
+from data_diff.errors import (
+    DataDiffCustomSchemaNoConfigError,
+    DataDiffDbtProjectVarsNotFoundError,
+    DataDiffNoAPIKeyError,
+    DataDiffNoDatasourceIdError,
+)
 
 from . import connect_to_table, diff_tables, Algorithm
-from .cloud import DatafoldAPI, TCloudApiDataDiff, TCloudApiOrgMeta, get_or_create_data_source
+from .cloud import DatafoldAPI, TCloudApiDataDiff, TCloudApiOrgMeta
 from .dbt_parser import DbtParser, TDatadiffConfig
 from .tracking import (
     bool_ask_for_email,
@@ -66,7 +71,6 @@ def dbt_diff(
     config = dbt_parser.get_datadiff_config()
     _initialize_events(dbt_parser.dbt_user_id, dbt_parser.dbt_version, dbt_parser.dbt_project_id)
 
-
     if not state and not (config.prod_database or config.prod_schema):
         doc_url = "https://docs.datafold.com/development_testing/open_source#configure-your-dbt-project"
         raise DataDiffDbtProjectVarsNotFoundError(
@@ -407,7 +411,7 @@ def _initialize_events(dbt_user_id: Optional[str], dbt_version: Optional[str], d
 
 
 def _email_signup() -> None:
-    email_regex = r'^[\w\.\+-]+@[\w\.-]+\.\w+$'
+    email_regex = r"^[\w\.\+-]+@[\w\.-]+\.\w+$"
     prompt = "\nWould you like to be notified when a new data-diff version is available?\n\nEnter email or leave blank to opt out (we'll only ask once).\n"
 
     if bool_ask_for_email():

data_diff/errors.py

@@ -49,8 +49,10 @@ class DataDiffDbtSelectVersionTooLowError(Exception):
 class DataDiffCustomSchemaNoConfigError(Exception):
     "Raised when a model has a custom schema, but there is no prod_custom_schema config. (And not using --state)."
 
+
 class DataDiffNoAPIKeyError(Exception):
     "Raised when using --cloud but no API key is present in the DATAFOLD_API_KEY env var or keyring"
 
+
 class DataDiffNoDatasourceIdError(Exception):
     "Raised when using --cloud but no datasource_id was found in dbt_project.yml"

tests/test_dbt.py

@@ -788,9 +788,7 @@ def test_diff_is_cloud_no_ds_id(
     @patch("data_diff.dbt._cloud_diff")
     @patch("data_diff.dbt_parser.DbtParser.__new__")
     @patch("builtins.input", return_value="n")
-    def test_diff_is_cloud_no_api_key(
-        self, _, mock_dbt_parser, mock_cloud_diff, mock_local_diff, mock_get_diff_vars
-    ):
+    def test_diff_is_cloud_no_api_key(self, _, mock_dbt_parser, mock_cloud_diff, mock_local_diff, mock_get_diff_vars):
         mock_dbt_parser_inst = Mock()
         mock_model = Mock()
         config = TDatadiffConfig(prod_database="prod_db", prod_schema="prod_schema")