fix: format/import

Author: bradhilton

dev/swebench/rollout.py

@@ -12,6 +12,7 @@
 from grpclib.exceptions import StreamTerminatedError
 from instances import Instance
 from langfuse.decorators import observe
+from logs import setup_agent_logger
 from pydantic import BaseModel
 from requests import adapters as requests_adapters
 from requests.exceptions import ConnectionError, ConnectTimeout, SSLError
@@ -26,7 +27,6 @@
 from urllib3.exceptions import ProtocolError
 
 import art
-from logs import setup_agent_logger
 from swebench.harness.modal_eval.run_evaluation_modal import app, run_instance_modal
 from swebench.harness.test_spec.test_spec import make_test_spec
 

src/art/local/backend.py

@@ -425,9 +425,9 @@ async def _train_model(
             num_gradient_steps = int(
                 result.pop("num_gradient_steps", estimated_gradient_steps)
             )
-            assert (
-                num_gradient_steps == estimated_gradient_steps
-            ), f"num_gradient_steps {num_gradient_steps} != estimated_gradient_steps {estimated_gradient_steps}"
+            assert num_gradient_steps == estimated_gradient_steps, (
+                f"num_gradient_steps {num_gradient_steps} != estimated_gradient_steps {estimated_gradient_steps}"
+            )
             results.append(result)
             yield {**result, "num_gradient_steps": num_gradient_steps}
             pbar.update(1)