Merge pull request #2228 from RTXteam/issue2139

Issue2139
RTXteam · Dec 14, 2023 · 34107b7 · 34107b7
2 parents 865f907 + 5cf93c3
commit 34107b7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/code/ARAX/ARAXQuery/Expand/kp_info_cacher.py b/code/ARAX/ARAXQuery/Expand/kp_info_cacher.py
@@ -7,7 +7,6 @@
 import pathlib
 import pickle
 import sys
-import psutil
 from datetime import datetime, timedelta
 from typing import Set, Dict, Optional
 
@@ -149,7 +148,6 @@ def load_kp_info_caches(self, log: ARAXResponse):
 
         except Exception as e:
             log.error(f"Unable to load KP info caches: {e}")
-            raise e
 
         # The caches MUST be up to date at this point, so we just load them
         log.debug(f"Loading cached Smart API amd meta map info")

diff --git a/code/ARAX/ARAXQuery/Expand/kp_selector.py b/code/ARAX/ARAXQuery/Expand/kp_selector.py
@@ -36,7 +36,11 @@ def _load_cached_kp_info(self) -> tuple:
         else:
             # Load cached KP info
             kp_cacher = KPInfoCacher()
-            smart_api_info, meta_map = kp_cacher.load_kp_info_caches(self.log)
+            try:
+                smart_api_info, meta_map = kp_cacher.load_kp_info_caches(self.log)
+            except Exception as e:
+                self.log.error(f"Failed to load KP info caches due to {e}", error_code="LoadKPCachesFailed")
+                return None, None, None, None
 
             # Record None URLs for our local KPs
             allowed_kp_urls = smart_api_info["allowed_kp_urls"]