11import json
22import os
33import tempfile
4- from pathlib import Path
54from unittest import mock
65
76import 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