Fix #4: Fix wandb generate_id crash — use string directly instead of int(id, 36)

itt_solver/wandb_runner.py

@@ -4,7 +4,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 import wandb
 
-from typing import Any, Dict
+from typing import Any, Dict, Optional, Union
 from itt_solver import experiment_driver as ed
 
 def _save_phi_image(phi: np.ndarray, path: str):
@@ -20,8 +20,8 @@ def run_and_log_wandb(task: Dict[str, Any],
                       params: Dict[str, Any],
                       out_dir: str = "experiments",
                       wandb_project: str = "itt_solver",
-                      wandb_entity: str | None = None,
-                      resume: bool | str = "allow"):
+                      wandb_entity: Optional[str] = None,
+                      resume: Union[bool, str] = "allow"):
     """
     Run a single experiment via experiment_driver.run_single and log results to W&B
     using a context manager (with wandb.init(...) as run: ...).
@@ -32,7 +32,7 @@ def run_and_log_wandb(task: Dict[str, Any],
     - out_dir: local directory where run_single writes artifacts (result.json, logs.json, phi_best.npy).
     - wandb_project: W&B project name.
     - wandb_entity: optional W&B entity (team/user).
-    - anonymous: "allow" to log anonymously if no API key is set, or False/None to require login.
+    - resume: "allow" to log anonymously if no API key is set, or False/None to require login.
     """
     # Run the experiment locally first (this writes files under out_dir)
     result = ed.run_single(task, atomic_library, params, out_dir)
@@ -54,11 +54,14 @@ def run_and_log_wandb(task: Dict[str, Any],
     result_path = os.path.join(out_dir, base + "_result.json") if base else None
     logs_path = os.path.join(out_dir, base + "_logs.json") if base else None
 
+    # Generate a safe run name using wandb's built-in ID generator
+    run_id = wandb.util.generate_id()
+
     # Start W&B run using context manager
     with wandb.init(project=wandb_project,
                     entity=wandb_entity,
                     config=params,
-                    name=f"{task.get('name','task')}_{int(wandb.util.generate_id(), 36)}",
+                    name=f"{task.get('name','task')}_{run_id}",
                     reinit=True,
                     resume=resume) as run: