ProcessGroup

Author: Borda

src/lightning/fabric/utilities/types.py

@@ -61,7 +61,7 @@ def load_state_dict(self, state_dict: dict[_DictKey, Any]) -> None: ...
 
 
 @runtime_checkable
-class CollectibleGroup(Protocol):
+class CollectibleGroup(ProcessGroup):
     def size(self) -> int: ...
 
     def rank(self) -> int: ...
@@ -81,7 +81,7 @@ def step(self, closure: Optional[Callable[[], float]] = ...) -> Optional[float]:
 
 
 @runtime_checkable
-class Optimizable(Steppable, Protocol):
+class Optimizable(Steppable):
     """To structurally type ``optimizer``"""
 
     param_groups: list[dict[Any, Any]]