From 949f4182023a443ca4b54661ea98948ac6223d42 Mon Sep 17 00:00:00 2001 From: "Zichen \"Charles\" Zhang" <52727818+zcczhang@users.noreply.github.com> Date: Tue, 20 Feb 2024 11:16:42 -0800 Subject: [PATCH] fix callback --- allenact/algorithms/onpolicy_sync/runner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/allenact/algorithms/onpolicy_sync/runner.py b/allenact/algorithms/onpolicy_sync/runner.py index 5302d198..bd884b90 100644 --- a/allenact/algorithms/onpolicy_sync/runner.py +++ b/allenact/algorithms/onpolicy_sync/runner.py @@ -240,17 +240,17 @@ def setup_callback_classes(self, callbacks: Optional[str]) -> Set[Callback]: callback_classes = [ mod_class[1] for mod_class in classes - if issubclass(mod_class[1], Callback) + if issubclass(mod_class[1], Callback) and mod_class[1] != Callback ] - assert callback_classes == 1, ( + assert len(callback_classes) == 1, ( f"Expected a single callback class in {filename}, but found {len(callback_classes)}." f" These classes were found: {callback_classes}." ) for mod_class in callback_classes: # NOTE: initialize the callback class - callback = mod_class[1]() + callback = mod_class() callback.setup(**setup_dict) callback_objects.add(callback)