From 45413455791c77465c5f33a5082053274eb18900 Mon Sep 17 00:00:00 2001 From: Stephan Hoyer Date: Wed, 19 Dec 2018 10:08:01 -0800 Subject: ENH: refactor __array_function__ pure Python implementation --- benchmarks/benchmarks/bench_overrides.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'benchmarks') diff --git a/benchmarks/benchmarks/bench_overrides.py b/benchmarks/benchmarks/bench_overrides.py index 2cb94c95c..b42c5d185 100644 --- a/benchmarks/benchmarks/bench_overrides.py +++ b/benchmarks/benchmarks/bench_overrides.py @@ -16,10 +16,10 @@ def mock_broadcast_to(array, shape, subok=False): def _concatenate_dispatcher(arrays, axis=None, out=None): - for array in arrays: - yield array if out is not None: - yield out + arrays = list(arrays) + arrays.append(out) + return arrays @array_function_dispatch(_concatenate_dispatcher) -- cgit v1.2.1