full import paths

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:

src/datacat_integration/operators.py

 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."""

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"