From 16174f5cc4761aef8004857b39d0a69c320391a5 Mon Sep 17 00:00:00 2001
From: Ivan Usov <ivan.usov@psi.ch>
Date: Thu, 10 Dec 2020 16:27:00 +0100
Subject: [PATCH] Fix zebra_mode dataset read

---
 pyzebra/h5.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyzebra/h5.py b/pyzebra/h5.py
index 69e7690..d0dd566 100644
--- a/pyzebra/h5.py
+++ b/pyzebra/h5.py
@@ -52,7 +52,10 @@ def read_detector_data(filepath):
 
         det_data = {"data": data}
 
-        det_data["zebra_mode"] = h5f.get("/entry1/zebra_mode", "nb")
+        if "/entry1/zebra_mode" in h5f:
+            det_data["zebra_mode"] = h5f["/entry1/zebra_mode"][0].decode()
+        else:
+            det_data["zebra_mode"] = "nb"
 
         # om, sometimes ph
         if det_data["zebra_mode"] == "nb":