Update trapz and in1d deprecation for NPY201

astral-sh · Jun 20, 2024 · 5fb9e50 · 5fb9e50
1 parent a26bd01
commit 5fb9e50
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/crates/ruff_linter/src/rules/numpy/rules/numpy_2_0_deprecation.rs b/crates/ruff_linter/src/rules/numpy/rules/numpy_2_0_deprecation.rs
@@ -306,6 +306,14 @@ pub(crate) fn numpy_2_0_deprecation(checker: &mut Checker, expr: &Expr) {
                     guideline: Some("Use the `np.errstate` context manager instead."),
                 },
             }),
+            ["numpy", "in1d"] => Some(Replacement {
+                existing: "in1d",
+                details: Details::AutoImport {
+                    path: "numpy",
+                    name: "isin",
+                    compatibility: Compatibility::BackwardsCompatible,
+                },
+            }),
             ["numpy", "INF"] => Some(Replacement {
                 existing: "INF",
                 details: Details::AutoImport {
@@ -536,6 +544,14 @@ pub(crate) fn numpy_2_0_deprecation(checker: &mut Checker, expr: &Expr) {
                     compatibility: Compatibility::BackwardsCompatible,
                 },
             }),
+            ["numpy", "trapz"] => Some(Replacement {
+                existing: "trapz",
+                details: Details::AutoImport {
+                    path: "numpy",
+                    name: "trapezoid",
+                    compatibility: Compatibility::Breaking,
+                },
+            }),
             ["numpy", "unicode_"] => Some(Replacement {
                 existing: "unicode_",
                 details: Details::AutoImport {