diff --git a/mara_pipelines/incremental_processing/file_dependencies.py b/mara_pipelines/incremental_processing/file_dependencies.py index 9e882ce..4469d90 100644 --- a/mara_pipelines/incremental_processing/file_dependencies.py +++ b/mara_pipelines/incremental_processing/file_dependencies.py @@ -5,7 +5,7 @@ from typing import List import sqlalchemy -from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import declarative_base import mara_db.dbs from .. import config diff --git a/mara_pipelines/incremental_processing/incremental_copy_status.py b/mara_pipelines/incremental_processing/incremental_copy_status.py index 64ac861..ffc5a74 100644 --- a/mara_pipelines/incremental_processing/incremental_copy_status.py +++ b/mara_pipelines/incremental_processing/incremental_copy_status.py @@ -3,7 +3,7 @@ from typing import List import sqlalchemy -from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import declarative_base import mara_db.config import mara_db.dbs diff --git a/mara_pipelines/incremental_processing/processed_files.py b/mara_pipelines/incremental_processing/processed_files.py index fae5247..6c2b503 100644 --- a/mara_pipelines/incremental_processing/processed_files.py +++ b/mara_pipelines/incremental_processing/processed_files.py @@ -4,7 +4,7 @@ from typing import Dict import sqlalchemy -from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import declarative_base import mara_db.config import mara_db.dbs diff --git a/mara_pipelines/logging/run_log.py b/mara_pipelines/logging/run_log.py index 94f78b7..6d16aef 100644 --- a/mara_pipelines/logging/run_log.py +++ b/mara_pipelines/logging/run_log.py @@ -3,7 +3,7 @@ from typing import List, Dict import sqlalchemy.orm -from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import declarative_base from .. import config from ..logging import pipeline_events, system_statistics diff --git a/setup.cfg b/setup.cfg index 3d12b51..cf69b34 100644 --- a/setup.cfg +++ b/setup.cfg @@ -20,6 +20,7 @@ install_requires = more-itertools>=3.1.0 psutil>=5.4.0 requests>=2.19.1 + SQLAlchemy>=1.4 [options.package_data] mara_pipelines = **/*.py, **/*.sql, ui/static/*