updated TrimmedLibrary for WeakPDELibrary case

dynamicslab · Dec 3, 2023 · 6642ca5 · 6642ca5
1 parent 58f3c53
commit 6642ca5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysindy/feature_library/base.py b/pysindy/feature_library/base.py
@@ -592,7 +592,7 @@ def transform(self, x_full):
         xp_full = []
         for x in x_full:
             xp = self.library.transform([x])[0]
-            xp = np.delete(xp, self.drop_inds, axis=x.ax_coord)
+            xp = np.delete(xp, self.drop_inds, axis=xp.ax_coord)
             xp = AxesArray(xp, comprehend_axes(xp))
             xp_full.append(xp)
         return xp_full