Hotfix: Linear longitudinal tracking accepts Qs...

...in the form of a 1D array with length of one, as provided by the PyHEADTAIL synchrotron.
PyCOMPLETE · Jun 13, 2016 · 7fccc08 · 7fccc08
1 parent da0b0b9
commit 7fccc08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trackers/simple_long_tracking.py b/trackers/simple_long_tracking.py
@@ -701,7 +701,7 @@ def __init__(self, alpha_array, circumference, Qs, D_x=0, D_y=0,*args, **kwargs)
         '''
         super(LinearMap, self).__init__(alpha_array, circumference,
                                         *args, **kwargs)
-        assert (np.isscalar(Qs)), "Qs has to be a scalar"
+        assert (len(np.atleast_1d(Qs)) == 1), "Qs can only have one entry!"
         self.Qs = Qs
         self.D_x = D_x
         self.D_y = D_y