Skip to content

Commit

Permalink
run pre-commit hook
Browse files Browse the repository at this point in the history
  • Loading branch information
ValentinaHutter committed Sep 13, 2024
1 parent 8fa7478 commit aae0705
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 7 deletions.
11 changes: 7 additions & 4 deletions openeo_processes_dask/process_implementations/arrays.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)


Expand Down
6 changes: 3 additions & 3 deletions tests/test_arrays.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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():
Expand Down

0 comments on commit aae0705

Please sign in to comment.