fixing the failures

zenml-io · Dec 16, 2024 · 47ed477 · 47ed477
1 parent b6d37e1
commit 47ed477
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/zenml/models/v2/core/artifact.py b/src/zenml/models/v2/core/artifact.py
@@ -241,7 +241,7 @@ def apply_sorting(
 
             query = query.add_columns(
                 latest_version_subquery.c.latest_version_created,
-            )
+            ).where(ArtifactSchema.id == latest_version_subquery.c.id)
 
             # Apply sorting based on the operand
             if operand == SorterOps.ASCENDING:

diff --git a/src/zenml/models/v2/core/model.py b/src/zenml/models/v2/core/model.py
@@ -382,7 +382,7 @@ def apply_sorting(
 
             query = query.add_columns(
                 latest_version_subquery.c.latest_version_created,
-            )
+            ).where(ModelSchema.id == latest_version_subquery.c.id)
 
             # Apply sorting based on the operand
             if operand == SorterOps.ASCENDING:

diff --git a/src/zenml/models/v2/core/pipeline.py b/src/zenml/models/v2/core/pipeline.py
@@ -376,7 +376,7 @@ def apply_sorting(
 
             query = query.add_columns(
                 latest_run_subquery.c.latest_run,
-            )
+            ).where(PipelineSchema.id == latest_run_subquery.c.id)
 
             if operand == SorterOps.ASCENDING:
                 query = query.order_by(