save: hotfix stripping sigma from 2D quantities. (#70)

* Try-Except 2-D array strip export. * Modify test.
dgbowl · Jan 27, 2023 · 88a2689 · 88a2689
1 parent f8f5adf
commit 88a2689
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/dgpost/utils/save.py b/src/dgpost/utils/save.py
@@ -53,7 +53,12 @@ def save(
     if not sigma:
         logger.warning(f"Stripping uncertainties from table.")
         for col in table.columns:
-            table[col] = unp.nominal_values(table[col].array)
+            try:
+                table[col] = unp.nominal_values(table[col].array)
+            except ValueError:
+                logger.warning(
+                    f"Cannot strip uncertainties from array quantity '{col}'."
+                )
 
     # find type of file in path or use default 'pkl'
     if type is None:

diff --git a/tests/test_realworld/ts1_reflection/recipe.yaml b/tests/test_realworld/ts1_reflection/recipe.yaml
@@ -43,4 +43,5 @@ transform:
         output: TM210
 save:
   - table: table
-    as: table.pkl
+    as: table.pkl
+    sigma: false
diff --git a/tests/test_realworld/ts1_reflection/ref.table.pkl b/tests/test_realworld/ts1_reflection/ref.table.pkl