Accept string

generate.py

@@ -26,12 +26,16 @@ def extract_code(completion: str) -> str:
     if code_block:
         return code_block.group(1)
 
-def get_solution_file_path(model: str, day: Optional[int] = None):
+def get_solution_file_path(model: str, day: Optional[str | int] = None):
     """Returns the path to the solution file for the given day and model. If day is None, returns the path to the solution file only."""
+
+    if isinstance(day, str):
+        day = int(day)
+    day_str = f"{day:02d}"
+
     if day is None:
         return f"solution_{model}.py"
 
-    day_str = f"{day:02d}"
     return f"day{day_str}/solution_{model}.py"