Small bugfix

Ensuring that the current change is a scalar. Some versions of numpy return a one-element array.
freegs-plasma · Feb 16, 2019 · 3647241 · 3647241
1 parent d05e439
commit 3647241
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/freegs/machine.py b/freegs/machine.py
@@ -695,7 +695,8 @@ def controlAdjust(self, current_change):
         controlcoils = [coil for label,coil in self.coils if coil.control]
 
         for coil, dI in zip(controlcoils, current_change):
-            coil.current += dI
+            # Ensure that dI is a scalar
+            coil.current += np.asscalar(dI)
 
     def controlCurrents(self):
         """