diff --git a/pyannote/audio/pipelines/utils/hook.py b/pyannote/audio/pipelines/utils/hook.py index fc6e56734..2a675d1c9 100644 --- a/pyannote/audio/pipelines/utils/hook.py +++ b/pyannote/audio/pipelines/utils/hook.py @@ -212,6 +212,13 @@ def __exit__(self, *args): if hasattr(hook, "__exit__"): hook.__exit__(*args) - def __call__(self, *args: Any, **kwds: Any) -> Any: + def __call__( + self, + step_name: Text, + step_artifact: Any, + file: Optional[Mapping] = None, + total: Optional[int] = None, + completed: Optional[int] = None, + ): for hook in self.hooks: - hook(*args, **kwds) + hook(step_name, step_artifact, file=file, total=total, completed=completed)