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