From aae0705b4eff36aea7233ed859d346435016abb0 Mon Sep 17 00:00:00 2001 From: ValentinaHutter Date: Fri, 13 Sep 2024 13:06:31 +0200 Subject: [PATCH] run pre-commit hook --- .../process_implementations/arrays.py | 11 +++++++---- tests/test_arrays.py | 6 +++--- 2 files changed, 10 insertions(+), 7 deletions(-) diff --git a/openeo_processes_dask/process_implementations/arrays.py b/openeo_processes_dask/process_implementations/arrays.py index 1d16ac44..c66b5546 100644 --- a/openeo_processes_dask/process_implementations/arrays.py +++ b/openeo_processes_dask/process_implementations/arrays.py @@ -214,16 +214,19 @@ def array_labels(data: ArrayLike) -> ArrayLike: return np.arange(len(data)) -def array_apply(data: ArrayLike, process: Callable, context: Optional[Any] = None) -> ArrayLike: +def array_apply( + data: ArrayLike, process: Callable, context: Optional[Any] = None +) -> ArrayLike: if not context: context = {} positional_parameters = {"x": 0} named_parameters = {"x": data, "context": context} if callable(process): process_to_apply = np.vectorize(process) - return process_to_apply(data, - positional_parameters = positional_parameters, - named_parameters = named_parameters, + return process_to_apply( + data, + positional_parameters=positional_parameters, + named_parameters=named_parameters, ) diff --git a/tests/test_arrays.py b/tests/test_arrays.py index 279bb4e3..0152ac09 100644 --- a/tests/test_arrays.py +++ b/tests/test_arrays.py @@ -10,11 +10,11 @@ from openeo_processes_dask.process_implementations.arrays import * from openeo_processes_dask.process_implementations.cubes.reduce import reduce_dimension -from openeo_processes_dask.process_implementations.math import add from openeo_processes_dask.process_implementations.exceptions import ( ArrayElementNotAvailable, TooManyDimensions, ) +from openeo_processes_dask.process_implementations.math import add from tests.general_checks import general_output_checks from tests.mockdata import create_fake_rastercube @@ -259,8 +259,8 @@ def test_array_apply(process_registry): x=ParameterReference(from_parameter="x"), ) - output_cube = array_apply(data=np.array([1,2,3,4,5,6]), process=_process) - assert (output_cube == [2,3,4,5,6,7]).all() + output_cube = array_apply(data=np.array([1, 2, 3, 4, 5, 6]), process=_process) + assert (output_cube == [2, 3, 4, 5, 6, 7]).all() def test_first():