fix: correct some typings

Author: yuxqiu

swebench/harness/test_spec.py

@@ -4,7 +4,7 @@
 import re
 
 from dataclasses import dataclass
-from typing import Any, Union
+from typing import Any, Union, cast
 
 from swebench.harness.constants import (
     SWEbenchInstance,
@@ -37,9 +37,9 @@ class TestSpec:
     instance_id: str
     repo: str
     version: str
-    repo_script_list: str
-    eval_script_list: str
-    env_script_list: str
+    repo_script_list: list[str]
+    eval_script_list: list[str]
+    env_script_list: list[str]
     arch: str
     FAIL_TO_PASS: list[str]
     PASS_TO_PASS: list[str]
@@ -104,15 +104,15 @@ def platform(self):
             return "linux/arm64/v8"
         else:
             raise ValueError(f"Invalid architecture: {self.arch}")
-        
+
 
 def get_test_specs_from_dataset(dataset: Union[list[SWEbenchInstance], list[TestSpec]]) -> list[TestSpec]:
     """
     Idempotent function that converts a list of SWEbenchInstance objects to a list of TestSpec objects.
     """
     if isinstance(dataset[0], TestSpec):
-        return dataset
-    return list(map(make_test_spec, dataset))
+        return cast(list[TestSpec], dataset)
+    return list(map(make_test_spec, cast(list[SWEbenchInstance], dataset)))
 
 
 def make_repo_script_list(specs, repo, repo_directory, base_commit, env_name):