From 551af4160bd505dcd09f49fad9a74eb77650a223 Mon Sep 17 00:00:00 2001 From: Doug Branton Date: Fri, 6 Oct 2023 09:01:58 -0700 Subject: [PATCH] drop brackets --- src/tape/ensemble.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py index 9bf60f90..c85b9174 100644 --- 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")