Update io.py

NVIDIA · Jan 10, 2025 · 9f9f3e5 · 9f9f3e5
1 parent a6e2fdd
commit 9f9f3e5
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/cosmos1/utils/io.py b/cosmos1/utils/io.py
@@ -31,10 +31,13 @@ def read_prompts_from_file(prompt_file: str) -> List[Dict[str, str]]:
         List[Dict[str, str]]: List of prompt dictionaries
     """
     prompts = []
-    with open(prompt_file, "r") as f:
-        for line in f:
-            prompt_dict = json.loads(line.strip())
-            prompts.append(prompt_dict)
+    try:
+        with open(prompt_file, "r") as f:
+            for line in f:
+                prompt_dict = json.loads(line.strip())
+                prompts.append(prompt_dict)
+    except (FileNotFoundError, json.JSONDecodeError) as e:
+        raise ValueError(f"Error reading prompts file: {e}")
     return prompts