From 9fd84e63bcc84c8dbd4c135564dbe7c86d510c03 Mon Sep 17 00:00:00 2001
From: dedebenui <41047847+dedebenui@users.noreply.github.com>
Date: Fri, 21 Jun 2024 08:08:00 +0200
Subject: [PATCH] Update `trapz` and `in1d` deprecation for NPY201 (#11948)

---
 .../rules/numpy/rules/numpy_2_0_deprecation.rs   | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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
index aaa735198a..62b3887811 100644
--- 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 {