Skip to content

Commit f281c7f

Browse files
committed
Fix lints
1 parent cdf5406 commit f281c7f

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

test/run/torchx_backend/schedulers/test_skypilot_jobs.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import json
22
import os
33
import tempfile
4-
from pathlib import Path
54
from unittest import mock
65

76
import pytest
@@ -157,8 +156,7 @@ def test_save_job_dir_new_file():
157156

158157
try:
159158
with mock.patch(
160-
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS",
161-
temp_path
159+
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS", temp_path
162160
):
163161
_save_job_dir("test_app_id", "RUNNING")
164162

@@ -182,8 +180,7 @@ def test_save_job_dir_existing_file():
182180

183181
try:
184182
with mock.patch(
185-
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS",
186-
temp_path
183+
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS", temp_path
187184
):
188185
_save_job_dir("new_app_id", "PENDING")
189186

@@ -212,8 +209,7 @@ def test_get_job_dirs_existing_file():
212209

213210
try:
214211
with mock.patch(
215-
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS",
216-
temp_path
212+
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS", temp_path
217213
):
218214
result = _get_job_dirs()
219215
assert result == test_data
@@ -227,8 +223,7 @@ def test_get_job_dirs_file_not_found():
227223
non_existent_path = "/tmp/definitely_does_not_exist_12345.json"
228224

229225
with mock.patch(
230-
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS",
231-
non_existent_path
226+
"nemo_run.run.torchx_backend.schedulers.skypilot_jobs.SKYPILOT_JOB_DIRS", non_existent_path
232227
):
233228
result = _get_job_dirs()
234229
assert result == {}

0 commit comments

Comments
 (0)