From 25d8818d4758cd49fe1a23f3fcfab52203212dee Mon Sep 17 00:00:00 2001
From: Fabian Joswig <fjosw@users.noreply.github.com>
Date: Sun, 8 Sep 2024 22:33:23 +0200
Subject: [PATCH 1/3] [Feat] Add support for index tricks in autograd numpy

---
 autograd/numpy/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/autograd/numpy/__init__.py b/autograd/numpy/__init__.py
index 28b3f7a8..121fd295 100644
--- a/autograd/numpy/__init__.py
+++ b/autograd/numpy/__init__.py
@@ -1,2 +1,7 @@
 from . import fft, linalg, numpy_boxes, numpy_jvps, numpy_vjps, numpy_vspaces, random
 from .numpy_wrapper import *
+from numpy import (
+    diag_indices_from, diag_indices, fill_diagonal, ndindex, ndenumerate,
+    ix_, c_, r_, s_, ogrid, mgrid, unravel_index, ravel_multi_index,
+    index_exp
+)

From b82e78edcc882b0dee7e3fb5b8ecf29dc926e44b Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Sun, 8 Sep 2024 20:35:58 +0000
Subject: [PATCH 2/3] style: pre-commit fixes

---
 autograd/numpy/__init__.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/autograd/numpy/__init__.py b/autograd/numpy/__init__.py
index 121fd295..6958e952 100644
--- a/autograd/numpy/__init__.py
+++ b/autograd/numpy/__init__.py
@@ -1,7 +1,19 @@
-from . import fft, linalg, numpy_boxes, numpy_jvps, numpy_vjps, numpy_vspaces, random
-from .numpy_wrapper import *
 from numpy import (
-    diag_indices_from, diag_indices, fill_diagonal, ndindex, ndenumerate,
-    ix_, c_, r_, s_, ogrid, mgrid, unravel_index, ravel_multi_index,
-    index_exp
+    c_,
+    diag_indices,
+    diag_indices_from,
+    fill_diagonal,
+    index_exp,
+    ix_,
+    mgrid,
+    ndenumerate,
+    ndindex,
+    ogrid,
+    r_,
+    ravel_multi_index,
+    s_,
+    unravel_index,
 )
+
+from . import fft, linalg, numpy_boxes, numpy_jvps, numpy_vjps, numpy_vspaces, random
+from .numpy_wrapper import *

From 237db4f1842afcf1043081e75e4ef81059f10039 Mon Sep 17 00:00:00 2001
From: Fabian Joswig <fjosw@users.noreply.github.com>
Date: Sun, 8 Sep 2024 22:59:39 +0200
Subject: [PATCH 3/3] [Feat] Remove redundant imports.

---
 autograd/numpy/__init__.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/autograd/numpy/__init__.py b/autograd/numpy/__init__.py
index 6958e952..bacf7381 100644
--- a/autograd/numpy/__init__.py
+++ b/autograd/numpy/__init__.py
@@ -1,17 +1,13 @@
 from numpy import (
-    c_,
     diag_indices,
     diag_indices_from,
     fill_diagonal,
     index_exp,
-    ix_,
     mgrid,
     ndenumerate,
     ndindex,
     ogrid,
-    r_,
     ravel_multi_index,
-    s_,
     unravel_index,
 )