Merge pull request #47 from AI4Bharat/audio_fix

AI4Bharat · Mar 19, 2024 · f2e64ac · f2e64ac
2 parents 30737b9 + 39127b2
commit f2e64ac
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/backend/projects/utils.py b/backend/projects/utils.py
@@ -23,20 +23,26 @@
 
 
 def get_audio_project_types():
-    with open("projects/project_registry.yaml") as f:
-        project_registry_details = yaml.load(f, Loader=SafeLoader)
+    try:
+        with open("projects/project_registry.yaml") as f:
+            project_registry_details = yaml.load(f, Loader=SafeLoader)
 
-    audio_project_types = project_registry_details["Audio"]["project_types"].keys()
+        audio_project_types = project_registry_details["Audio"]["project_types"].keys()
+    except Exception as e:
+        return []
     return audio_project_types
 
 
 def get_translation_dataset_project_types():
-    with open("projects/project_registry.yaml") as f:
-        project_registry_details = yaml.load(f, Loader=SafeLoader)
+    try:
+        with open("projects/project_registry.yaml") as f:
+            project_registry_details = yaml.load(f, Loader=SafeLoader)
 
-    translation_project_types = project_registry_details["Translation"][
-        "project_types"
-    ].keys()
+        translation_project_types = project_registry_details["Translation"][
+            "project_types"
+        ].keys()
+    except Exception as e:
+        return []
     return translation_project_types