Update src/safeds/data/tabular/containers/_tagged_table.py

Co-authored-by: Simon Breuer <[email protected]>
Safe-DS · Jul 8, 2023 · ff3782b · ff3782b
1 parent 0db7131
commit ff3782b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/safeds/data/tabular/containers/_tagged_table.py b/src/safeds/data/tabular/containers/_tagged_table.py
@@ -412,12 +412,11 @@ def keep_only_columns(self, column_names: list[str]) -> TaggedTable:
             raise IllegalSchemaModificationError("Must keep the target column.")
         if len(set(self.features.column_names).intersection(set(column_names))) == 0:
             raise IllegalSchemaModificationError("Must keep at least one feature column.")
-        table = super().keep_only_columns(column_names)
         return TaggedTable._from_table(
-            table,
+            super().keep_only_columns(column_names),
             target_name=self.target.name,
             feature_names=sorted(
-                set(self.features.column_names).intersection(set(table.column_names)),
+                set(self.features.column_names).intersection(set(column_names)),
                 key={val: ix for ix, val in enumerate(self.features.column_names)}.__getitem__,
             ),
         )