Replace "math" with "numpy" library for optimise

freegs-plasma · Nov 13, 2023 · 52189ad · 52189ad
1 parent d562e32
commit 52189ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/freegs/optimise.py b/freegs/optimise.py
@@ -27,7 +27,7 @@
 import matplotlib.pyplot as plt
 from freegs.plotting import plotEquilibrium
 
-from math import sqrt
+from numpy import sqrt, inf
 
 # Measures which operate on Equilibrium objects
 
@@ -199,7 +199,7 @@ def solve_and_measure(eq):
             return measure(eq)
         except:
             # Solve failed.
-            return float("inf")
+            return float(inf)
 
     # Call the generic optimiser,
     return optimiser.optimise(