Merge pull request #76 from tbody-cfs/fix-ndarray-item

Replace deprecated np.asscalar
freegs-plasma · Dec 14, 2022 · 937c01f · 937c01f
2 parents 4e233b7 + 1163b58
commit 937c01f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/freegs/equilibrium.py b/freegs/equilibrium.py
@@ -349,7 +349,7 @@ def q(self, psinorm=None, npsi=100):
         result = critical.find_safety(self, psinorm=psinorm)
         # Convert to a scalar if only one result
         if len(result) == 1:
-            return np.asscalar(result)
+            return result[0]
         return result
 
     def tor_flux(self, psi=None):
@@ -366,7 +366,7 @@ def tor_flux(self, psi=None):
 
         # Convert to a scalar if only one result
         if len(result) == 1:
-            return np.asscalar(result)
+            return result[0]
         return result
 
     def rhotor(self, psi=None):
@@ -388,7 +388,7 @@ def rhotor(self, psi=None):
         result = np.sqrt(torflux / max_torflux)
 
         if len(result) == 1:
-            return np.asscalar(result)
+            return result[0]
         return result
 
     def pprime(self, psinorm):