Make selection_mod optional without breaking positional args

Author: RolandBERTINJOHANNET

shimmer/modules/global_workspace.py

@@ -723,7 +723,6 @@ def __init__(
         workspace_dim: int,
         loss_coefs: BroadcastLossCoefs | Mapping[str, float],
         selection_temperature: float = 0.2,
-        selection_mod: SelectionBase | None = None,
         optim_lr: float = 1e-3,
         optim_weight_decay: float = 0.0,
         scheduler_args: SchedulerArgs | None = None,
@@ -733,6 +732,7 @@ def __init__(
         | None
         | OneCycleSchedulerSentinel = OneCycleSchedulerSentinel.DEFAULT,
         fusion_activation_fn: Callable[[torch.Tensor], torch.Tensor] = torch.tanh,
+        selection_mod: SelectionBase | None = None,
     ) -> None:
         """
         Initializes a Global Workspace
@@ -766,6 +766,8 @@ def __init__(
                 no scheduler will be used. Defaults to use OneCycleScheduler
             fusion_activation_fn (`Callable[[torch.Tensor], torch.Tensor]`): activation
                 function to fuse the domains.
+            selection_mod (`SelectionBase | None`): optional custom selection module.
+                If None (default), uses `RandomSelection`.
         """
         domain_mods = freeze_domain_modules(domain_mods)
         gw_mod = GWModule(