Fix lint issues

Author: shawnlewis

programmer/evals/eval_repeated_edits.py

@@ -10,14 +10,14 @@
 
 from ..agent import AgentState, Agent
 from ..config import agent, agent_claude, agent_claude_replace, agent_replace
-from ..tools import tool_context
+from ..tools import tool_context, LocalToolContext
 
 
 # @pytest.fixture
 @contextmanager
 def tempdir():
     with tempfile.TemporaryDirectory() as dir_:
-        with tool_context(dir_) as tc:
+        with tool_context(LocalToolContext(dir_)) as tc:
             yield tc
 
 

programmer/swebench/ingest-existing/make_dataset.py

@@ -1,6 +1,6 @@
 import argparse
 import sys
-
+from typing import Optional
 import pandas as pd
 import weave
 
@@ -18,13 +18,13 @@ def load_raw_dataset(name: str, split: str):
     )
 
 
-def load_weave_dataset(name: str, split: str, limit: int = None):
+def load_weave_dataset(name: str, split: str, limit: Optional[int] = None):
     df = load_raw_dataset(name, split)
 
     data_list = df.to_dict("records")
     data_list = data_list[:limit] if limit else data_list
 
-    return weave.Dataset(name=f"Verified-{limit}", rows=data_list)
+    return weave.Dataset(name=f"Verified-{limit}", rows=data_list)  # type: ignore
 
 
 def main(dataset_name="SWE-bench_Verified", split="test"):
@@ -48,7 +48,7 @@ def main(dataset_name="SWE-bench_Verified", split="test"):
 
     data_list = df.to_dict("records")
 
-    dataset = weave.Dataset(rows=data_list)
+    dataset = weave.Dataset(rows=data_list)  # type: ignore
 
     weave.publish(dataset, f"{dataset_name}_{split}")
 

programmer/swebench/ingest/make_dataset.py

@@ -38,7 +38,7 @@ def main():
     pivot_df = pivot_df[cols]
 
     # Sort the pivot table by 'models_resolved_true' in descending order
-    pivot_df = pivot_df.sort_values(by="models_resolved_true", ascending=False)
+    pivot_df = pivot_df.sort_values(by="models_resolved_true", ascending=False)  # type: ignore
 
     # Sort columns by the model that got the most resolved
     model_success_count = pivot_df.sum().sort_values(ascending=False)