mypy fix

cleanlab · Aug 1, 2023 · 2603868 · 2603868
1 parent 24b4733
commit 2603868
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cleanlab_studio/internal/api/api.py b/cleanlab_studio/internal/api/api.py
@@ -1,8 +1,7 @@
 import io
 import os
 import time
-from itertools import chain
-from typing import Callable, List, Optional, Tuple, Dict, Union, Any
+from typing import Callable, cast, List, Optional, Tuple, Dict, Union, Any
 from cleanlab_studio.errors import APIError
 
 import requests
@@ -371,4 +370,4 @@ def get_prediction_status(api_key: str, query_id: str) -> Dict[str, str]:
     )
     handle_api_error(res)
 
-    return res.json()
+    return cast(dict, res.json())