From fe580b57b4fe0e35e3d5145801faa00b991d485c Mon Sep 17 00:00:00 2001
From: caufieldjh <j.harry.caufield@gmail.com>
Date: Fri, 23 Aug 2024 16:31:05 -0400
Subject: [PATCH] Type hint fixes

---
 src/ontogpt/utils/multilingual.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ontogpt/utils/multilingual.py b/src/ontogpt/utils/multilingual.py
index 36c3b4a20..b26e7a903 100644
--- 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)