From 54892318b94f1f20cc0e45f8b0e74818892a45a0 Mon Sep 17 00:00:00 2001
From: Michael Schuster <michael.schuster.ffb@googlemail.com>
Date: Fri, 13 Dec 2024 09:37:30 +0100
Subject: [PATCH] Linting

---
 src/zenml/zen_stores/migrations/utils.py | 4 ++--
 src/zenml/zen_stores/sql_zen_store.py    | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/zenml/zen_stores/migrations/utils.py b/src/zenml/zen_stores/migrations/utils.py
index 58598e292d..b6e42413a0 100644
--- a/src/zenml/zen_stores/migrations/utils.py
+++ b/src/zenml/zen_stores/migrations/utils.py
@@ -317,8 +317,8 @@ def backup_database_to_storage(
                         data = []
                         for row in rows:
                             row_dict = row._asdict()
-                            for column in ignore_columns:
-                                row_dict.pop(column, None)
+                            for c in ignore_columns:
+                                row_dict.pop(c, None)
 
                             data.append(row_dict)
 
diff --git a/src/zenml/zen_stores/sql_zen_store.py b/src/zenml/zen_stores/sql_zen_store.py
index 990b7ac519..6d80083fe2 100644
--- a/src/zenml/zen_stores/sql_zen_store.py
+++ b/src/zenml/zen_stores/sql_zen_store.py
@@ -71,6 +71,7 @@
     col,
     create_engine,
     delete,
+    desc,
     or_,
     select,
 )