From 2fe6001e57b86f12e5c1620e2c0daa707bb78dd1 Mon Sep 17 00:00:00 2001 From: Dan Date: Fri, 28 Apr 2023 11:32:23 -0600 Subject: [PATCH] Clarify dependency error --- data_diff/dbt.py | 11 ----------- data_diff/dbt_parser.py | 2 +- 2 files changed, 1 insertion(+), 12 deletions(-) diff --git a/data_diff/dbt.py b/data_diff/dbt.py index 1f4cd01d..392c984d 100644 --- a/data_diff/dbt.py +++ b/data_diff/dbt.py @@ -18,17 +18,6 @@ logger = getLogger(__name__) -def import_dbt(): - try: - from dbt_artifacts_parser.parser import parse_run_results, parse_manifest - from dbt.config.renderer import ProfileRenderer - import yaml - except ImportError: - raise RuntimeError("Could not import 'dbt' package. You can install it using: pip install 'data-diff[dbt]'.") - - return parse_run_results, parse_manifest, ProfileRenderer, yaml - - from .tracking import ( set_entrypoint_name, set_dbt_user_id, diff --git a/data_diff/dbt_parser.py b/data_diff/dbt_parser.py index 77f9fe70..55f7e395 100644 --- a/data_diff/dbt_parser.py +++ b/data_diff/dbt_parser.py @@ -18,7 +18,7 @@ def import_dbt(): from dbt.config.renderer import ProfileRenderer import yaml except ImportError: - raise RuntimeError("Could not import 'dbt' package. You can install it using: pip install 'data-diff[dbt]'.") + raise RuntimeError("Please run pip install 'data-diff[dbt]' to install missing package dependencies.") return parse_run_results, parse_manifest, ProfileRenderer, yaml