drop brackets

lincc-frameworks · Oct 6, 2023 · 551af41 · 551af41
1 parent e6b6d38
commit 551af41
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py
@@ -423,7 +423,7 @@ def assign(self, table="object", temporary=False, **kwargs):
             post_cols = self._object.columns
 
             if temporary:
-                self._object_temp.extend([col for col in post_cols if col not in pre_cols])
+                self._object_temp.extend(col for col in post_cols if col not in pre_cols)
 
         elif table == "source":
             pre_cols = self._source.columns
@@ -432,7 +432,7 @@ def assign(self, table="object", temporary=False, **kwargs):
             post_cols = self._source.columns
 
             if temporary:
-                self._source_temp.extend([col for col in post_cols if col not in pre_cols])
+                self._source_temp.extend(col for col in post_cols if col not in pre_cols)
 
         else:
             raise ValueError(f"{table} is not one of 'object' or 'source'")
@@ -576,7 +576,7 @@ def calc_nobs(self, by_band=False, label="nobs", temporary=True):
             self._object = self._object.assign(**{label + "_" + band: band_counts[band] for band in bands})
 
             if temporary:
-                self._object_temp.extend([label + "_" + band for band in bands])
+                self._object_temp.extend(label + "_" + band for band in bands)
 
         else:
             counts = self._source.groupby([self._id_col])[[self._band_col]].aggregate("count")