fix nits

cleanlab_studio/internal/api/api.py

@@ -363,7 +363,7 @@ def start_prediction(api_key: str, model_id: str, query_id: str) -> None:
 
 
 def get_prediction_status(api_key: str, query_id: str) -> Dict[str, str]:
-    """Gets status of model prediction query. Returns status, and optionally the result_url or error message."""
+    """Gets status of model prediction query. Returns status, and optionally the result url or error message."""
     res = requests.get(
         f"{model_base_url}/predict/{query_id}",
         headers=_construct_headers(api_key),

cleanlab_studio/studio/inference.py

@@ -2,14 +2,13 @@
 import csv
 import io
 import time
-from typing import List, Optional, Tuple, Union
+from typing import List, Tuple, Union
 from typing_extensions import TypeAlias
 
 import numpy as np
 import numpy.typing as npt
 import pandas as pd
 
-from cleanlab_studio.errors import APIError
 from cleanlab_studio.internal.api import api