def METHOD_NAME(self, mocked_weights: WeightsEnum) -> None:
    BYOLTask(
        model=mocked_weights.meta["model"],
        weights=str(mocked_weights),
        in_channels=mocked_weights.meta["in_chans"],
    )