fix: weight can be float or int

Loop3D · Nov 8, 2021 · d2c469f · d2c469f
1 parent 9b487fd
commit d2c469f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/LoopStructural/interpolators/discrete_interpolator.py b/LoopStructural/interpolators/discrete_interpolator.py
@@ -179,7 +179,7 @@ def add_constraints_to_least_squares(self, A, B, idc, w = 1., name='undefined'):
         mask = np.any(np.isnan(A),axis=1)
         A[mask,:] = 0
         A[length>0,:] /= length[length>0,None]
-        if isinstance(w,float):
+        if isinstance(w,(float,int)):
             w = np.ones(A.shape[0])*w
 
         if isinstance(w,np.ndarray) == False: