Skip to content

Commit

Permalink
hide mypy type check error on load_from_chekpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
bdvllrs committed Jul 30, 2024
1 parent 529ae63 commit c7d5554
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion shimmer/modules/global_workspace.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit c7d5554

Please sign in to comment.