return None for PCA

mostly-ai · Jan 20, 2025 · 35ce379 · 35ce379
1 parent cfbc578
commit 35ce379
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mostlyai/qa/_filesystem.py b/mostlyai/qa/_filesystem.py
@@ -288,6 +288,8 @@ def store_pca_model(self, pca_model: PCA):
         sio.dump(pca_model, self.pca_model_path)
 
     def load_pca_model(self) -> PCA | None:
+        if not self.pca_model_path.exists():
+            return None
         unknown_types = sio.get_untrusted_types(file=self.pca_model_path)
         if unknown_types:
             raise ValueError(f"Unknown types found in file {self.pca_model_path}: {unknown_types}")