fix: typetracer nplike's all function leads to infinite recursion for axis=None (#3765)

* fix typetracer nplikes all function

* add test

* instantiate nplike only once in test

---------

Co-authored-by: Peter Fackeldey <fackeldey.peter@gmail.com>

Author: ikrommyd

src/awkward/_nplikes/typetracer.py

@@ -1559,7 +1559,7 @@ def all(
         if axis is None:
             return self.all(
                 cast(TypeTracerArray, self.reshape(x, (-1,))),
-                axis=axis,
+                axis=0,
                 keepdims=keepdims,
                 maybe_out=maybe_out,
             )

tests/test_3765_typetracer_all_any_infinite_recursion.py

@@ -0,0 +1,37 @@
+# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
+
+from __future__ import annotations
+
+import numpy as np
+
+from awkward._nplikes.typetracer import TypeTracer, TypeTracerArray
+
+nplike = TypeTracer.instance()
+
+
+def test_all():
+    buffer = TypeTracerArray._new(np.dtype("float32"), (4,))
+    result = nplike.all(buffer, axis=None)
+    assert isinstance(result, TypeTracerArray)
+    assert result.dtype == np.dtype("bool")
+    assert result.shape == ()
+
+    buffer = TypeTracerArray._new(np.dtype("float32"), (4, 3))
+    result = nplike.all(buffer, axis=None)
+    assert isinstance(result, TypeTracerArray)
+    assert result.dtype == np.dtype("bool")
+    assert result.shape == ()
+
+
+def test_any():
+    buffer = TypeTracerArray._new(np.dtype("float32"), (4,))
+    result = nplike.any(buffer, axis=None)
+    assert isinstance(result, TypeTracerArray)
+    assert result.dtype == np.dtype("bool")
+    assert result.shape == ()
+
+    buffer = TypeTracerArray._new(np.dtype("float32"), (4, 3))
+    result = nplike.any(buffer, axis=None)
+    assert isinstance(result, TypeTracerArray)
+    assert result.dtype == np.dtype("bool")
+    assert result.shape == ()