From e5ab1b4cc4c87f68dc5a0065b2346090edacbc40 Mon Sep 17 00:00:00 2001 From: Jacob Date: Thu, 9 Jan 2025 21:55:45 -0600 Subject: [PATCH] fix lower/upper vs std --- torchcast/__init__.py | 2 +- torchcast/state_space/predictions.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/torchcast/__init__.py b/torchcast/__init__.py index 2b8877c..93b60a1 100644 --- a/torchcast/__init__.py +++ b/torchcast/__init__.py @@ -1 +1 @@ -__version__ = '0.5.0' +__version__ = '0.5.1' diff --git a/torchcast/state_space/predictions.py b/torchcast/state_space/predictions.py index 7dd6239..0e49e82 100644 --- a/torchcast/state_space/predictions.py +++ b/torchcast/state_space/predictions.py @@ -463,7 +463,11 @@ def _tensor_to_df(tens, measures): raise ValueError("Expected `type` to be 'predictions' or 'components'.") out = pd.concat(out).reset_index(drop=True) - _out_cols = [group_colname, time_colname, 'measure', 'mean', 'lower', 'upper'] + _out_cols = [group_colname, time_colname, 'measure', 'mean'] + if conf is None: + _out_cols.append('std') + else: + _out_cols.extend(['lower', 'upper']) if type.startswith('comp'): _out_cols = _out_cols[0:3] + ['process', 'state_element'] + _out_cols[3:] if 'actual' in out.columns: