diff --git a/src/datacat_integration/hooks.py b/src/datacat_integration/hooks.py index 185f5cec401a8c054c64b754f01402ed8bbf238a..542da6ef279721015a7362f92a9a6d81acf087fc 100644 --- a/src/datacat_integration/hooks.py +++ b/src/datacat_integration/hooks.py @@ -7,9 +7,9 @@ from requests.sessions import session from requests import Session from typing import Optional, Dict -from connection import DataCatConnection, DataCatalogEntry +from datacat_integration.connection import DataCatConnection, DataCatalogEntry -from auth import BearerAuth +from datacat_integration.auth import BearerAuth def get_connection_from_entry(data: Dict[str, Any], datacat_type: str, oid: str) -> Connection: diff --git a/src/datacat_integration/operators.py b/src/datacat_integration/operators.py index 4cb97640c74057c5840a34ca087842fba604b909..718ee3b01831eba1898d2e289f896e85e04dfee8 100644 --- a/src/datacat_integration/operators.py +++ b/src/datacat_integration/operators.py @@ -1,5 +1,5 @@ from airflow.models.baseoperator import BaseOperator -from hooks import DataCatalogHook +from datacat_integration.hooks import DataCatalogHook class GetDatacatalogEntryOperator(BaseOperator): """This task returns the data for an entry in the datacatalog.""" diff --git a/src/datacat_integration/secrets.py b/src/datacat_integration/secrets.py index d57f0b4a26b2e318ce8abee642ca740a9841d055..7f389d0e0ef9da1a6ffbd4072fbe154feb125f49 100644 --- a/src/datacat_integration/secrets.py +++ b/src/datacat_integration/secrets.py @@ -3,9 +3,9 @@ from urllib.parse import urljoin from airflow.secrets import BaseSecretsBackend from airflow.models.connection import Connection import requests -from connection import DataCatConnection -from hooks import DataCatalogHook, get_connection_from_entry +from datacat_integration.connection import DataCatConnection +from datacat_integration.hooks import DataCatalogHook, get_connection_from_entry connection_backend_type = "airflow_connections"