TDEngineTarget: Keep table_col if present in columns or tag_cols (

#523)
mlrun · May 16, 2024 · 1fb5462 · 1fb5462
1 parent 0193af2
commit 1fb5462
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/storey/targets.py b/storey/targets.py
@@ -866,7 +866,7 @@ def __init__(
 
         if table_col:
             kwargs["key_field"] = table_col
-            if kwargs.get("drop_key_field") is None:
+            if kwargs.get("drop_key_field") is None and table_col not in columns and table_col not in tag_cols:
                 kwargs["drop_key_field"] = True
 
         _Batching.__init__(self, **kwargs)