Replace rows on conflict

WordPress · Sep 7, 2024 · 3c8c2b8 · 3c8c2b8
1 parent d0ad001
commit 3c8c2b8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/catalog/dags/data_augmentation/rekognition/add_rekognition_labels.py b/catalog/dags/data_augmentation/rekognition/add_rekognition_labels.py
@@ -55,7 +55,12 @@ def _insert_tags(tags_buffer: types.TagsBuffer, postgres_conn_id: str):
         postgres_conn_id=postgres_conn_id,
         default_statement_timeout=constants.INSERT_TIMEOUT,
     )
-    postgres.insert_rows(constants.TEMP_TABLE_NAME, tags_buffer, executemany=True)
+    postgres.insert_rows(
+        constants.TEMP_TABLE_NAME,
+        tags_buffer,
+        executemany=True,
+        replace=True,
+    )
 
 
 @task(trigger_rule=TriggerRule.NONE_FAILED_MIN_ONE_SUCCESS)