diff --git a/covalent_dispatcher/_db/datastore.py b/covalent_dispatcher/_db/datastore.py index a4734591d..2fe99015b 100644 --- a/covalent_dispatcher/_db/datastore.py +++ b/covalent_dispatcher/_db/datastore.py @@ -19,6 +19,7 @@ from pathlib import Path from typing import Generator, Optional +from alembic import command from alembic.config import Config from alembic.environment import EnvironmentContext from alembic.migration import MigrationContext @@ -27,7 +28,6 @@ from sqlalchemy.orm import Session, sessionmaker from sqlalchemy_utils import create_database, database_exists -from alembic import command from covalent._shared_files.config import get_config from . import models diff --git a/covalent_migrations/env.py b/covalent_migrations/env.py index ee8e5b7db..5c55d7847 100644 --- a/covalent_migrations/env.py +++ b/covalent_migrations/env.py @@ -16,9 +16,9 @@ from logging.config import fileConfig +from alembic import context from sqlalchemy import engine_from_config, pool -from alembic import context from covalent_dispatcher._db.datastore import DataStore from covalent_dispatcher._db.models import Base diff --git a/covalent_migrations/versions/1142d81b29b8_schema_updates_for_new_dal.py b/covalent_migrations/versions/1142d81b29b8_schema_updates_for_new_dal.py index 727d88ad0..795711bbb 100644 --- a/covalent_migrations/versions/1142d81b29b8_schema_updates_for_new_dal.py +++ b/covalent_migrations/versions/1142d81b29b8_schema_updates_for_new_dal.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic. diff --git a/covalent_migrations/versions/26d71848a404_v12.py b/covalent_migrations/versions/26d71848a404_v12.py index ddf64ac39..5b23a7756 100644 --- a/covalent_migrations/versions/26d71848a404_v12.py +++ b/covalent_migrations/versions/26d71848a404_v12.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic. diff --git a/covalent_migrations/versions/97202f5f47cb_v13.py b/covalent_migrations/versions/97202f5f47cb_v13.py index 96430bcd4..021884e44 100644 --- a/covalent_migrations/versions/97202f5f47cb_v13.py +++ b/covalent_migrations/versions/97202f5f47cb_v13.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic. diff --git a/covalent_migrations/versions/9b9d58f02985_v11.py b/covalent_migrations/versions/9b9d58f02985_v11.py index 16f3d8595..83781dd35 100644 --- a/covalent_migrations/versions/9b9d58f02985_v11.py +++ b/covalent_migrations/versions/9b9d58f02985_v11.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic. diff --git a/covalent_migrations/versions/b60c5ecdf927_init.py b/covalent_migrations/versions/b60c5ecdf927_init.py index 4d7b572c3..f4bf1bde1 100644 --- a/covalent_migrations/versions/b60c5ecdf927_init.py +++ b/covalent_migrations/versions/b60c5ecdf927_init.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic. diff --git a/covalent_migrations/versions/de0a6c0a3e3d_add_qelectron_data_exists_flag_to_db.py b/covalent_migrations/versions/de0a6c0a3e3d_add_qelectron_data_exists_flag_to_db.py index 789460ac4..b1a904814 100644 --- a/covalent_migrations/versions/de0a6c0a3e3d_add_qelectron_data_exists_flag_to_db.py +++ b/covalent_migrations/versions/de0a6c0a3e3d_add_qelectron_data_exists_flag_to_db.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic. diff --git a/covalent_migrations/versions/f64ecaa040d5_add_jobs_db_table.py b/covalent_migrations/versions/f64ecaa040d5_add_jobs_db_table.py index 1450982f8..6e412a322 100644 --- a/covalent_migrations/versions/f64ecaa040d5_add_jobs_db_table.py +++ b/covalent_migrations/versions/f64ecaa040d5_add_jobs_db_table.py @@ -22,7 +22,6 @@ """ import sqlalchemy as sa - from alembic import op # revision identifiers, used by Alembic.