weco 0.2.9__py3-none-any.whl → 0.2.11__py3-none-any.whl

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/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.dist-info → weco-0.2.11.dist-info}/METADATA

@@ -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.11.dist-info/RECORD

@@ -0,0 +1,11 @@
+weco/__init__.py,sha256=FRQQiNgU5a_qo5d4NNNgMXZ5YvwxlvISipBZeXH4wjc,125
+weco/api.py,sha256=89lB2572jApAxkA0DDppDnJKBwvZTa3kH9jFpC0LFDQ,3313
+weco/cli.py,sha256=fisH7Gvhupu-mgpC4Dmt4x5y0KtVo_o-LjYU_iTymW4,18481
+weco/panels.py,sha256=R_df-VAbWyLoqCA9A6UzbIGZ9sm2IgJO4idnyjmrHQk,12701
+weco/utils.py,sha256=hhIebUPnetFMfNSFfcsKVw1TSpeu_Zw3rBPPnxDie0U,3911
+weco-0.2.11.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
+weco-0.2.11.dist-info/METADATA,sha256=XqDYBWUfDcLowdk1Ycib5SKLHOi4vQ8Ogbt6M1go7Wk,9379
+weco-0.2.11.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+weco-0.2.11.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.2.11.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.2.11.dist-info/RECORD,,

weco-0.2.9.dist-info/RECORD

@@ -1,11 +0,0 @@
-weco/__init__.py,sha256=8vfBL3OdlOveq9nYktaI3JQLXAeR7Pnwz8TRZ3xu0nA,124
-weco/api.py,sha256=89lB2572jApAxkA0DDppDnJKBwvZTa3kH9jFpC0LFDQ,3313
-weco/cli.py,sha256=iTecBQ0gAqyQixj7tNU9S6f9NW7TMSGZtH8nw-85_Oc,18276
-weco/panels.py,sha256=R_df-VAbWyLoqCA9A6UzbIGZ9sm2IgJO4idnyjmrHQk,12701
-weco/utils.py,sha256=hhIebUPnetFMfNSFfcsKVw1TSpeu_Zw3rBPPnxDie0U,3911
-weco-0.2.9.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
-weco-0.2.9.dist-info/METADATA,sha256=Fq37xE2H8Ia3brOv6AcjsleqDeDL_U6XiV3bJHZOhCY,9378
-weco-0.2.9.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-weco-0.2.9.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.2.9.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.2.9.dist-info/RECORD,,