Don't run migration for empty updates (#3210)

zenml-io · Nov 25, 2024 · 39f9ca9 · 39f9ca9
1 parent f45f739
commit 39f9ca9
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/zenml/zen_stores/migrations/versions/ec6307720f92_simplify_model_version_links.py b/src/zenml/zen_stores/migrations/versions/ec6307720f92_simplify_model_version_links.py
@@ -59,12 +59,13 @@ def _migrate_artifact_type() -> None:
                 {"id_": artifact_version_id, "type": "ServiceArtifact"}
             )
 
-    connection.execute(
-        sa.update(artifact_version_table).where(
-            artifact_version_table.c.id == sa.bindparam("id_")
-        ),
-        updates,
-    )
+    if updates:
+        connection.execute(
+            sa.update(artifact_version_table).where(
+                artifact_version_table.c.id == sa.bindparam("id_")
+            ),
+            updates,
+        )
 
 
 def upgrade() -> None: