bugfix when existing idx is empty

epoch8 · Nov 2, 2024 · a855a83 · a855a83
1 parent 8055860
commit a855a83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datapipe/meta/sql_meta.py b/datapipe/meta/sql_meta.py
@@ -234,7 +234,7 @@ def get_existing_idx(self, idx: Optional[IndexDF] = None) -> IndexDF:
                 # Empty index -> empty result
                 return cast(
                     IndexDF,
-                    pd.DataFrame(columns=[column.name for column in self.sql_schema]),  # type: ignore
+                    pd.DataFrame(columns=self.primary_keys),  # type: ignore
                 )
             idx_cols = list(set(idx.columns.tolist()) & set(self.primary_keys))
         else: