fix: fix of list of Hooks

pyannote · Oct 22, 2023 · ec4eb02 · ec4eb02
1 parent 4b09aef
commit ec4eb02
Showing 1 changed file with 9 additions and 2 deletions.
diff --git 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)