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)
|
||||
|
||||
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
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user