(tmpdir, num_checkpoints)
| 10 | |
| 11 | @pytest.mark.parametrize('num_checkpoints', [1, 2, 12, 24]) |
| 12 | def test_get_optim_files(tmpdir, num_checkpoints): |
| 13 | saved_files = [] |
| 14 | for i in range(num_checkpoints): |
| 15 | file_name = "zero_" + str(i) + "_optim_states.pt" |
| 16 | path_name = os.path.join(tmpdir, file_name) |
| 17 | saved_files.append(path_name) |
| 18 | with open(path_name, "w") as f: |
| 19 | f.write(file_name) |
| 20 | loaded_files = get_optim_files(tmpdir) |
| 21 | for lf, sf in zip(loaded_files, saved_files): |
| 22 | assert lf == sf |
nothing calls this directly
no test coverage detected
searching dependent graphs…