Decode Axis Names (#390)

* Decode Axis Names * Changelog
wright-group · Jul 19, 2021 · 2f966a5 · 2f966a5
1 parent d356dd6
commit 2f966a5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -13,6 +13,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/).
 - axes with whitespace don't necessarily fail
 - Motortune Tunepoints only calculated if the checkbox is checked
 - old "identity" behavior removed which caused splitting on "F" in axis names
+- Decode when needed for axis dropdown in plot tab
 
 ### Added
 - Widgets to control slit widths and mirror positions of monochromator

diff --git a/yaqc_cmds/_plot.py b/yaqc_cmds/_plot.py
@@ -99,7 +99,12 @@ def on_channels_changed(self):
 
     def on_data_file_created(self):
         with somatic._wt5.data_container as data:
-            allowed = [x.split("{")[0].strip() for x in data.attrs["axes"]]
+            try:
+                allowed = [x.decode().split("{")[0].strip() for x in data.attrs["axes"]]
+            except AttributeError:
+                allowed = [x.split("{")[0].strip() for x in data.attrs["axes"]]
+            except:
+                allowed = [None]
             if "wa" in allowed:
                 allowed.remove("wa")
             self.axis.set_allowed_values(allowed)