Merge branch 'schema-updates' into add-trial-runner-id-column

microsoft · Jan 4, 2025 · b6c867c · b6c867c
2 parents 4e99658 + 908b5db
commit b6c867c
Show file tree

Hide file tree

Showing 8 changed files with 12 additions and 3 deletions.
diff --git a/conda-envs/mlos-3.10.yml b/conda-envs/mlos-3.10.yml
@@ -28,6 +28,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4

diff --git a/conda-envs/mlos-3.11.yml b/conda-envs/mlos-3.11.yml
@@ -28,6 +28,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4

diff --git a/conda-envs/mlos-3.12.yml b/conda-envs/mlos-3.12.yml
@@ -30,6 +30,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4

diff --git a/conda-envs/mlos-3.13.yml b/conda-envs/mlos-3.13.yml
@@ -30,6 +30,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4

diff --git a/conda-envs/mlos-windows.yml b/conda-envs/mlos-windows.yml
@@ -31,6 +31,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4

diff --git a/conda-envs/mlos.yml b/conda-envs/mlos.yml
@@ -26,6 +26,7 @@ dependencies:
     - pre-commit==4.0.1
     - pycodestyle==2.12.1
     - pylint==3.3.3
+    - tomlkit
     - mypy==1.14.1
     - pandas-stubs
     - types-beautifulsoup4

diff --git a/mlos_bench/mlos_bench/launcher.py b/mlos_bench/mlos_bench/launcher.py
@@ -380,7 +380,10 @@ def add_argument(self, *args: Any, **kwargs: Any) -> None:
             default=False,
             dest="create_update_storage_schema_only",
             action="store_true",
-            help="Makes sure that the storage schema is update to date for the current version of mlos_bench.",
+            help=(
+                "Makes sure that the storage schema is up to date "
+                "for the current version of mlos_bench."
+            ),
         )
 
         # By default we use the command line arguments, but allow the caller to

diff --git a/mlos_bench/mlos_bench/storage/sql/alembic/versions/d2a708351ba8_add_alembic.py b/mlos_bench/mlos_bench/storage/sql/alembic/versions/d2a708351ba8_add_alembic.py
@@ -11,8 +11,8 @@
 """
 from collections.abc import Sequence
 
-import sqlalchemy as sa
-from alembic import op
+# import sqlalchemy as sa
+# from alembic import op
 
 # revision identifiers, used by Alembic.
 revision: str = "d2a708351ba8"