fix monitoring preventing operation kwargs for processing fn getting forwarded

This commit is contained in:
Julius Unverfehrt 2023-03-28 17:13:59 +02:00
parent 793a427c50
commit 3c17047377

View File

@ -35,11 +35,11 @@ class PrometheusMonitor:
return self._add_result_monitoring(process_fn)
def _add_result_monitoring(self, process_fn: Callable):
def inner(data: Any):
def inner(data: Any, **kwargs):
start = time()
result: Sized = process_fn(data)
result: Sized = process_fn(data, **kwargs)
runtime = time() - start