Set imat = np.ones((1, 1), dtype=float) if self.greville.size == 1

pyccel · Oct 1, 2024 · b462135 · b462135
1 parent 5ff3884
commit b462135
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/psydac/fem/splines.py b/psydac/fem/splines.py
@@ -171,14 +171,17 @@ def init_interpolation( self, dtype=float ):
         Greville points.
 
         """
-        imat = collocation_matrix(
-            knots    = self.knots,
-            degree   = self.degree,
-            periodic = self.periodic,
-            normalization = self.basis,
-            xgrid    = self.greville,
-            multiplicity = self.multiplicity
-        )
+        if self.greville.size == 1:
+            imat = np.ones((1, 1), dtype=float)
+        else:
+            imat = collocation_matrix(
+                knots    = self.knots,
+                degree   = self.degree,
+                periodic = self.periodic,
+                normalization = self.basis,
+                xgrid    = self.greville,
+                multiplicity = self.multiplicity
+            )
 
         if self.periodic:
             # Convert to CSC format and compute sparse LU decomposition