fix lower/upper vs std

strongio · Jan 10, 2025 · f33c638 · f33c638
1 parent f339fef
commit f33c638
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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: