chore: update paths

app.py

@@ -56,10 +56,10 @@ ALL_DIRECTORIES = [
 # Load test dataset
 
 print("Load data ...")
-UNIQUE_FOUNDERS = load_pickle_from_zip("unique_founders.pkl")
-OTHER_TEST_FOUNDERS = load_pickle_from_zip("hf_test_founders.pkl")
-OTHER_TRAIN_FOUNDERS = load_pickle_from_zip("hf_train_founders.pkl")
-UNIQUE_MIXED_FOUNDERS = load_pickle_from_zip("unique_mixed_founders.pkl")
+UNIQUE_FOUNDERS = load_pickle_from_zip("data/unique_founders.pkl")
+OTHER_TEST_FOUNDERS = load_pickle_from_zip("data/hf_test_founders.pkl")
+OTHER_TRAIN_FOUNDERS = load_pickle_from_zip("data/hf_train_founders.pkl")
+UNIQUE_MIXED_FOUNDERS = load_pickle_from_zip("data/unique_mixed_founders.pkl")
 
 DESCENDANT_PATH = Path("./data/Child.pkl")
 PREDICTION_IMG_PATH = Path("output.npg")

utils.py

@@ -39,15 +39,21 @@ META = {"A": 5, "C": 1059079, "M": 10589, "NW": 100, "CT": 1059, "CTR": 0.1, "WS
 
 BUILD_GENS = [1, 2, 4, 6, 8, 12, 16, 24, 32, 48]
 
+import os 
 
 def load_pickle_from_zip(file_name, zip_path="data.zip"):
     """
     Load a pickle file from within a zip archive.
     """
+    if not os.path.exists(zip_path):
+        raise FileNotFoundError(f"The zip file '{zip_path}' does not exist.")
+    
     with zipfile.ZipFile(zip_path, 'r') as z:
+        if file_name not in z.namelist():
+            print("-----", file_name, z.namelist())
+            raise KeyError(f"The file '{file_name}' does not exist in the zip archive '{zip_path}'.")
         with z.open(file_name) as f:
             return pickle.load(f)
-            
 
 def generate_weighted_percentages():
     dominant_percentage = random.randint(50, 70)