From c7d55540e78be3db3817e4283d12770ab6085aec Mon Sep 17 00:00:00 2001 From: bdvllrs Date: Tue, 30 Jul 2024 12:44:39 +0000 Subject: [PATCH] hide mypy type check error on load_from_chekpoint --- shimmer/modules/global_workspace.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,