fix monitoring preventing operation kwargs for processing fn getting forwarded
This commit is contained in:
parent
793a427c50
commit
3c17047377
@ -35,11 +35,11 @@ class PrometheusMonitor:
|
|||||||
return self._add_result_monitoring(process_fn)
|
return self._add_result_monitoring(process_fn)
|
||||||
|
|
||||||
def _add_result_monitoring(self, process_fn: Callable):
|
def _add_result_monitoring(self, process_fn: Callable):
|
||||||
def inner(data: Any):
|
def inner(data: Any, **kwargs):
|
||||||
|
|
||||||
start = time()
|
start = time()
|
||||||
|
|
||||||
result: Sized = process_fn(data)
|
result: Sized = process_fn(data, **kwargs)
|
||||||
|
|
||||||
runtime = time() - start
|
runtime = time() - start
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user