Selaa lähdekoodia

Adding check for the file to see if it exists

Beto 1 vuosi sitten
vanhempi
commit
fd7041aa64
1 muutettua tiedostoa jossa 2 lisäystä ja 0 poistoa
  1. 2 0
      tests/test_train_utils.py

+ 2 - 0
tests/test_train_utils.py

@@ -113,4 +113,6 @@ def test_save_to_json(temp_output_dir, mocker):
     )
 
     assert results["metrics_filename"] not in ["", None]
+    assert os.path.isfile(results["metrics_filename"])
+