fix unit test

nxexox · Oct 19, 2023 · 8cf870f · 8cf870f
1 parent 34903cb
commit 8cf870f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit_tests/ml/test_storage.py b/tests/unit_tests/ml/test_storage.py
@@ -101,8 +101,8 @@ def test_load_many_from_folder(self, pickle_with_x_model, pickle_with_x_model_tw
         # Order in folder
         assert str(models_bin[0].path) in {str(pickle_with_x_model), str(pickle_with_x_model_two)}
         assert models_bin[0].raw_data in {with_x_model_data, with_x_model_data_two}
-        assert str(models_bin[1].path) == {str(pickle_with_x_model), str(pickle_with_x_model_two)}
-        assert models_bin[1].raw_data == {with_x_model_data, with_x_model_data_two}
+        assert str(models_bin[1].path) in {str(pickle_with_x_model), str(pickle_with_x_model_two)}
+        assert models_bin[1].raw_data in {with_x_model_data, with_x_model_data_two}
 
     def test_load_single_file_by_not_default_mask(self, joblib_print_model):
         storage = DiskStorage(