moving to a new api for polyfit and ensure the absemce of nans and infs in the input

Author: dkazanc

httomolibgpu/recon/rotation.py

@@ -21,6 +21,7 @@
 """Modules for finding the axis of rotation for 180 and 360 degrees scans"""
 
 import numpy as np
+from numpy.polynomial import Polynomial
 from httomolibgpu import cupywrapper
 
 cp = cupywrapper.cp
@@ -712,9 +713,23 @@ def _calculate_curvature(list_metric):
 
     # work mostly on CPU here - we have very small arrays here
     list1 = cp.asnumpy(list_metric[min_pos - radi : min_pos + radi + 1])
-    afact1 = np.polyfit(np.arange(0, 2 * radi + 1), list1, 2)[0]
+    list1[np.isnan(list1)] = list1[~np.isnan(list1)].mean()
+    list1[np.isinf(list1)] = list1[~np.isinf(list1)].mean()
+
+    # afact1 = np.polyfit(np.arange(0, 2 * radi + 1), list1, 2)[0]
+
+    series1 = Polynomial.fit(np.arange(0, 2 * radi + 1), list1, deg=2)
+    afact1 = series1.convert().coef[-1]
+
     list2 = cp.asnumpy(list_metric[min_pos - 1 : min_pos + 2])
-    (afact2, bfact2, _) = np.polyfit(np.arange(min_pos - 1, min_pos + 2), list2, 2)
+    list2[np.isnan(list2)] = list2[~np.isnan(list2)].mean()
+    list2[np.isinf(list2)] = list2[~np.isinf(list2)].mean()
+
+    # (afact2, bfact2, _) = np.polyfit(np.arange(min_pos - 1, min_pos + 2), list2, 2)
+
+    series2 = Polynomial.fit(np.arange(min_pos - 1, min_pos + 2), list2, deg=2)
+    afact2 = series2.convert().coef[-1]
+    bfact2 = series2.convert().coef[-1 - 1]
 
     curvature = np.abs(afact1)
     if afact2 != 0.0: