Merge pull request #391 from NannyML/fix/filter-univariate-result

Optimize filtering univariate result for period

nannyml/base.py

@@ -293,13 +293,15 @@ def _filter(
         *args,
         **kwargs,
     ) -> Self:
+        res = super()._filter(period, *args, **kwargs)
+        if metrics is None and column_names is None:
+            return res
+
         if metrics is None:
             metrics = [metric.column_name for metric in self.metrics]
         if column_names is None:
             column_names = self.column_names
 
-        res = super()._filter(period, *args, **kwargs)
-
         data = pd.concat([res.data.loc[:, (['chunk'])], res.data.loc[:, (column_names, metrics)]], axis=1)
         data = data.reset_index(drop=True)