def METHOD_NAME(self) -> None:
    c = Config(["test.py", "--gpus", "1", "-j", "2"])
    s = FakeSystem()
    stage = m.GPU(c, s)
    assert stage.spec.workers == 2
    assert (
        stage.spec.shards
        == [
            Shard([(0,)]),
            Shard([(1,)]),
            Shard([(2,)]),
            Shard([(3,)]),
            Shard([(4,)]),
            Shard([(5,)]),
        ]
        * stage.spec.workers
    )