Use proxies for Array API's valid_tuple_axes

hypothesis-python/src/hypothesis/extra/array_api.py

@@ -15,7 +15,6 @@
 
 import math
 from collections import defaultdict
-from copy import deepcopy
 from functools import update_wrapper, wraps
 from numbers import Real
 from types import SimpleNamespace
@@ -50,6 +49,7 @@
     valid_tuple_axes as _valid_tuple_axes,
 )
 from hypothesis.internal.conjecture import utils as cu
+from hypothesis.internal.reflection import proxies
 from hypothesis.internal.validation import (
     check_type,
     check_valid_bound,
@@ -684,14 +684,17 @@ def floating_dtypes(
     return st.sampled_from(dtypes)
 
 
-# TODO: prevent NumPy from mutating the doc
-valid_tuple_axes = deepcopy(_valid_tuple_axes)
+@proxies(_valid_tuple_axes)
+def valid_tuple_axes(*args, **kwargs):
+    return _valid_tuple_axes(*args, **kwargs)
+
+
 valid_tuple_axes.__doc__ = f"""
     Return a strategy for permissible tuple-values for the ``axis``
     argument in Array API sequential methods e.g. ``sum``, given the specified
     dimensionality.
 
-    {valid_tuple_axes.__doc__}
+    {_valid_tuple_axes.__doc__}
     """
 
 
@@ -715,7 +718,6 @@ def mutually_broadcastable_shapes(
     )
 
 
-# TODO: prevent NumPy from mutating the doc
 mutually_broadcastable_shapes.__doc__ = _mutually_broadcastable_shapes.__doc__