fix: handle exception when loading cache file in test_inference.py (m…

…icrosoft#5802) This PR is to fix CI failures such as https://github.com/microsoft/DeepSpeed/actions/runs/10085903860/job/27887546470#step:8:3616 cc @tjruwase Co-authored-by: Logan Adams <[email protected]>
samadejacobs · Jul 29, 2024 · 58241b1 · 58241b1
1 parent afe1b9e
commit 58241b1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/inference/test_inference.py b/tests/unit/inference/test_inference.py
@@ -94,7 +94,10 @@ def _hf_model_list() -> List[ModelInfo]:
     model_data = {"cache_time": 0, "model_list": []}
     if os.path.isfile(cache_file_path):
         with open(cache_file_path, 'rb') as f:
-            model_data = pickle.load(f)
+            try:
+                model_data = pickle.load(f)
+            except Exception as e:
+                print(f"Error loading cache file {cache_file_path}: {e}")
 
     current_time = time.time()