From 5a340f2d46085bf025368899f6e0fc0e690714b7 Mon Sep 17 00:00:00 2001 From: Maxime Armstrong Date: Fri, 21 Jun 2024 13:00:03 -0400 Subject: [PATCH] Update DbtProject --- .../libraries/dagster-dbt/dagster_dbt/dbt_project.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python_modules/libraries/dagster-dbt/dagster_dbt/dbt_project.py b/python_modules/libraries/dagster-dbt/dagster_dbt/dbt_project.py index 4c3bea2e80364..eac838838cd2f 100644 --- a/python_modules/libraries/dagster-dbt/dagster_dbt/dbt_project.py +++ b/python_modules/libraries/dagster-dbt/dagster_dbt/dbt_project.py @@ -25,7 +25,7 @@ def using_dagster_dev() -> bool: class DbtManifestPreparer: """A dbt manifest represented by DbtProject.""" - def ensure_prepared(self, project: "DbtProject") -> None: + def on_load(self, project: "DbtProject") -> None: """Invoked when DbtProject is instantiated with this preparer.""" def prepare(self, project: "DbtProject") -> None: @@ -55,7 +55,7 @@ def __init__( """ self._generate_cli_args = generate_cli_args or ["parse", "--quiet"] - def ensure_prepared(self, project: "DbtProject"): + def on_load(self, project: "DbtProject"): if self.using_dagster_dev() or self.parse_on_load_opt_in(): self.prepare(project) if not project.manifest_path.exists(): @@ -197,7 +197,7 @@ def __new__( if not using_dagster_dev() and packaged_project_dir and packaged_project_dir.exists(): project_dir = packaged_project_dir - manifest_preparer = DagsterDbtManifestPreparer() + preparer = DagsterDbtManifestPreparer() manifest_path = project_dir.joinpath(target_path, "manifest.json")