Type hint fixes

monarch-initiative · Aug 23, 2024 · fe580b5 · fe580b5
1 parent 030256c
commit fe580b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ontogpt/utils/multilingual.py b/src/ontogpt/utils/multilingual.py
@@ -31,8 +31,8 @@ def multilingual_analysis(
 
     # Keep track of the predictions
     # Key is the filename, value is a list of predictions
-    pred_ids = {}
-    pred_names = {}
+    pred_ids: dict[str, list[str]] = {}
+    pred_names: dict[str, list[str]] = {}
 
     try:
         gpt_diagnosis = ai.complete(prompt)