Fix imports

Author: skinnerjc

src/inspect_evals/gdm_capabilities/intercode_ctf/task.py

@@ -3,13 +3,22 @@
 
 from inspect_ai import Task, task
 from inspect_ai.scorer import includes
-from inspect_ai.solver import Solver, basic_agent, solver, system_message
+from inspect_ai.solver import (
+    Generate,
+    Solver,
+    TaskState,
+    basic_agent,
+    chain,
+    solver,
+    system_message,
+)
 from inspect_ai.tool import bash, python
 
 from .dataset import read_dataset
 
 COMPOSE_FILE = Path(__file__).parent / "compose.yaml"
 
+
 @solver
 def prepend_flag_format() -> Solver:
     """A solver that prepends the expected flag format to the task input message."""