Python LSFEnvironment.node_rank Examples

Example #1

File: test_lsf_environment.py Project: krishnakalyan3/pytorch-lightning

def test_node_rank(tmp_path):
    environ = {
        "LSB_DJOB_RANKFILE": _make_rankfile(tmp_path),
        "LSB_JOBID": "1234",
        "JSM_NAMESPACE_SIZE": "4",
        "JSM_NAMESPACE_RANK": "3",
        "JSM_NAMESPACE_LOCAL_RANK": "1",
    }
    with mock.patch.dict(os.environ,
                         environ), mock.patch("socket.gethostname",
                                              return_value="10.10.10.2"):
        env = LSFEnvironment()
        assert env.node_rank() == 2

Example #2

File: test_lsf_environment.py Project: thoglu/pytorch-lightning

def test_node_rank(_):
    env = LSFEnvironment()
    assert env.node_rank() == 2