fix: disable alembic migration

RafalSiwek · Dec 25, 2023 · e17828e · e17828e
1 parent 2275d97
commit e17828e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/charm.py b/src/charm.py
@@ -95,6 +95,7 @@ def _get_env_vars(self, relational_db_data):
         ret_env_vars = {
             "ZENML_STORE_TYPE": "sql",
             "ZENML_STORE_URL": f"mysql://{relational_db_data['username']}:{relational_db_data['password']}@{relational_db_data['host']}:{relational_db_data['port']}/{self._database_name}",  # noqa: E501
+            "DISABLE_DATABASE_MIGRATION": "True", # To avoid migrations https://github.com/zenml-io/zenml/tree/9a69295e9aabaa90156b0cc6115f585a91d0f108/src/zenml/zen_stores/migrations # noqa: E501
             "ZENML_STORE_SSL_VERIFY_SERVER_CERT": "false",
             "ZENML_SERVER_DEPLOYMENT_TYPE": "kubernetes",
             "ZENML_DEFAULT_PROJECT_NAME": "default",

diff --git a/tests/unit/test_operator.py b/tests/unit/test_operator.py
@@ -35,6 +35,7 @@
 EXPECTED_ENVIRONMENT = {
     "ZENML_STORE_TYPE": "sql",
     "ZENML_STORE_URL": "mysql://username:password@host:port/zenml",
+    "DISABLE_DATABASE_MIGRATION": "True",
     "ZENML_STORE_SSL_VERIFY_SERVER_CERT": "false",
     "ZENML_SERVER_DEPLOYMENT_TYPE": "kubernetes",
     "ZENML_DEFAULT_PROJECT_NAME": "default",