Merge pull request #395 from lnls-sirius/PR-fixbug-devices

Fix bug in devices.tune
lnls-sirius · Mar 10, 2020 · eb18403 · eb18403
2 parents bebc46a + 5e9b083
commit eb18403
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/siriuspy/siriuspy/VERSION b/siriuspy/siriuspy/VERSION
@@ -1 +1 @@
-1.19.0
+1.19.1
diff --git a/siriuspy/siriuspy/devices/tune.py b/siriuspy/siriuspy/devices/tune.py
@@ -70,11 +70,14 @@ class DEVICES:
 
     def __init__(self, devname):
         """."""
-        if devname == Tune.DEVICES.ALL:
-            tune_frac_h = TuneFrac(TuneFrac.DEVICES.SI_H)
-            tune_frac_v = TuneFrac(TuneFrac.DEVICES.SI_V)
-            tune_proc_h = TuneProc(TuneProc.DEVICES.SI_H)
-            tune_proc_v = TuneProc(TuneProc.DEVICES.SI_V)
+        # check if device exists
+        if devname not in Tune.DEVICES.ALL:
+            raise NotImplementedError(devname)
+
+        tune_frac_h = TuneFrac(TuneFrac.DEVICES.SI_H)
+        tune_frac_v = TuneFrac(TuneFrac.DEVICES.SI_V)
+        tune_proc_h = TuneProc(TuneProc.DEVICES.SI_H)
+        tune_proc_v = TuneProc(TuneProc.DEVICES.SI_V)
 
         devices = (tune_frac_h, tune_frac_v, tune_proc_h, tune_proc_v)