diff --git a/shimmer/modules/global_workspace.py b/shimmer/modules/global_workspace.py index 8def76c5..e3ae6ab4 100644 --- a/shimmer/modules/global_workspace.py +++ b/shimmer/modules/global_workspace.py @@ -864,7 +864,7 @@ def pretrained_global_workspace( gw_mod, selection_mod, domain_mods, loss_coefs, contrastive_fn ) - gw = GlobalWorkspace2Domains.load_from_checkpoint( + gw = GlobalWorkspace2Domains.load_from_checkpoint( # type: ignore[operator] checkpoint_path, gw_mod=gw_mod, selection_mid=selection_mod,