chore: Enable SIM Ruff rule (#442)

MeltanoLabs · Sep 23, 2024 · ab7a5e8 · ab7a5e8
1 parent f27557e
commit ab7a5e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -95,6 +95,7 @@ select = [
   "UP",  # pyupgrade
   "ICN", # flake8-import-conventions
   "RET", # flake8-return
+  "SIM", # flake8-simplify
   "RUF", # ruff
 ]
 

diff --git a/target_postgres/sinks.py b/target_postgres/sinks.py
@@ -237,7 +237,7 @@ def upsert(
             # Update
             where_condition = join_condition
             update_columns = {}
-            for column_name in self.schema["properties"].keys():
+            for column_name in self.schema["properties"]:
                 from_table_column: sa.Column = from_table.columns[column_name]
                 to_table_column: sa.Column = to_table.columns[column_name]
                 update_columns[to_table_column] = from_table_column