ensure write to file for each step

Author: DhruvSrikanth

.github/workflows/release.yml

@@ -90,7 +90,7 @@ jobs:
         GITHUB_TOKEN: ${{ github.token }}
       run: >-
         gh release create
-        'v0.2.1'
+        'v0.2.2'
         --repo '${{ github.repository }}'
         --notes ""
 
@@ -102,5 +102,5 @@ jobs:
       # sigstore-produced signatures and certificates.
       run: >-
         gh release upload
-        'v0.2.1' dist/**
+        'v0.2.2' dist/**
         --repo '${{ github.repository }}'

pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 ]
 description = "Documentation for `weco`, a CLI for using Weco AI's code optimizer."
 readme = "README.md"
-version = "0.2.1"
+version = "0.2.2"
 license = {text = "MIT"}
 requires-python = ">=3.12"
 dependencies = ["requests", "rich"]

weco/__init__.py

@@ -1,4 +1,4 @@
 # DO NOT EDIT
-__pkg_version__ = "0.2.1"
+__pkg_version__ = "0.2.2"
 __api_version__ = "v1"
 __base_url__ = f"https://api.aide.weco.ai/{__api_version__}"

weco/cli.py

@@ -185,6 +185,9 @@ def main() -> None:
                 # Save next solution (.runs/<session-id>/step_<step>.py)
                 write_to_path(fp=runs_dir / f"step_{step}.py", content=eval_and_next_solution_response["code"])
 
+                # Write the next solution to the source file
+                write_to_path(fp=source_fp, content=eval_and_next_solution_response["code"])
+
                 # Get the optimization session status for
                 # the best solution, its score, and the history to plot the tree
                 status_response = get_optimization_session_status(console=console, session_id=session_id, include_history=True)