weco 0.2.9__tar.gz → 0.2.11__tar.gz

{weco-0.2.9 → weco-0.2.11}/.github/workflows/release.yml

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

{weco-0.2.9 → weco-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.9
+Version: 0.2.11
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT

{weco-0.2.9 → weco-0.2.11}/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.9"
+version = "0.2.11"
 license = {text = "MIT"}
 requires-python = ">=3.8"
 dependencies = ["requests", "rich"]

{weco-0.2.9 → weco-0.2.11}/weco/__init__.py

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

{weco-0.2.9 → weco-0.2.11}/weco/cli.py

@@ -125,7 +125,7 @@ def main() -> None:
             runs_dir.mkdir(parents=True, exist_ok=True)
 
             # Save the original code (.runs/<session-id>/original.<extension>)
-            runs_copy_source_fp = runs_dir / f"original.{source_fp.suffix}"
+            runs_copy_source_fp = runs_dir / f"original{source_fp.suffix}"
             write_to_path(fp=runs_copy_source_fp, content=source_code)
 
             # Write the code string to the source file path
@@ -342,8 +342,8 @@ def main() -> None:
             # Update the end optimization layout
             final_message = (
                 f"{summary_panel.metric_name.capitalize()} {'maximized' if summary_panel.maximize else 'minimized'}! Best solution {summary_panel.metric_name.lower()} = [green]{status_response['best_result']['metric_value']}[/] 🏆"
-                if best_solution_node is not None
-                else "[red] No solution found.[/]"
+                if best_solution_node is not None and best_solution_node.metric is not None
+                else "[red] No valid solution found.[/]"
             )
             end_optimization_layout["summary"].update(summary_panel.get_display(final_message=final_message))
             end_optimization_layout["tree"].update(tree_panel.get_display())
@@ -352,8 +352,13 @@ def main() -> None:
             # Save optimization results
             # If the best solution does not exist or is has not been measured at the end of the optimization
             # save the original solution as the best solution
-            best_solution_code = best_solution_node.code
-            best_solution_score = best_solution_node.metric
+            if best_solution_node is not None:
+                best_solution_code = best_solution_node.code
+                best_solution_score = best_solution_node.metric
+            else:
+                best_solution_code = None
+                best_solution_score = None
+
             if best_solution_code is None or best_solution_score is None:
                 best_solution_content = (
                     f"# Weco could not find a better solution\n\n{read_from_path(fp=runs_copy_source_fp, is_json=False)}"
@@ -368,7 +373,7 @@ def main() -> None:
                 best_solution_content = f"# Best solution from Weco with a score of {best_score_str}\n\n{best_solution_code}"
 
             # Save best solution to .runs/<session-id>/best.<extension>
-            write_to_path(fp=runs_dir / f"best.{source_fp.suffix}", content=best_solution_content)
+            write_to_path(fp=runs_dir / f"best{source_fp.suffix}", content=best_solution_content)
 
             # write the best solution to the source file
             if not args.preserve_source:

{weco-0.2.9 → weco-0.2.11}/weco.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.9
+Version: 0.2.11
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT