diff --git a/ingestion/setup.py b/ingestion/setup.py index 250242ff748..96c426aef8f 100644 --- a/ingestion/setup.py +++ b/ingestion/setup.py @@ -185,7 +185,7 @@ plugins: Dict[str, Set[str]] = { "google-cloud", VERSIONS["boto3"], VERSIONS["google-cloud-storage"], - "dbt-artifacts-parser", + "collate-dbt-artifacts-parser", VERSIONS["azure-storage-blob"], VERSIONS["azure-identity"], }, @@ -363,7 +363,7 @@ test = { "pytest-order", "dirty-equals", # install dbt dependency - "dbt-artifacts-parser", + "collate-dbt-artifacts-parser", "freezegun", VERSIONS["sqlalchemy-databricks"], VERSIONS["databricks-sdk"], diff --git a/ingestion/src/metadata/ingestion/source/database/dbt/dbt_service.py b/ingestion/src/metadata/ingestion/source/database/dbt/dbt_service.py index ec6ea35c97e..40c5e96d9a8 100644 --- a/ingestion/src/metadata/ingestion/source/database/dbt/dbt_service.py +++ b/ingestion/src/metadata/ingestion/source/database/dbt/dbt_service.py @@ -15,7 +15,7 @@ DBT service Topology. from abc import ABC, abstractmethod from typing import Iterable, List -from dbt_artifacts_parser.parser import ( +from collate_dbt_artifacts_parser.parser import ( parse_catalog, parse_manifest, parse_run_results, diff --git a/ingestion/tests/unit/test_dbt.py b/ingestion/tests/unit/test_dbt.py index ffe24f33728..46022a87bd5 100644 --- a/ingestion/tests/unit/test_dbt.py +++ b/ingestion/tests/unit/test_dbt.py @@ -8,7 +8,11 @@ from pathlib import Path from unittest import TestCase from unittest.mock import MagicMock, patch -from dbt_artifacts_parser.parser import parse_catalog, parse_manifest, parse_run_results +from collate_dbt_artifacts_parser.parser import ( + parse_catalog, + parse_manifest, + parse_run_results, +) from pydantic import AnyUrl from metadata.generated.schema.entity.data.table import Column, DataModel, Table