Merge pull request #144 from foundation-model-stack/fix_int_program_id

fixed issue where program_id was int when should have been string

Author: JRosenkranz

scripts/drive_paged_programs.py

@@ -440,12 +440,12 @@ def parse_program_limit(limit_str: str) -> tuple[int, str]:
 
     for program_str in args.programs:
         enforce_prompt_split = program_str.split(":")
+        program_id = enforce_prompt_split[0]
         if len(enforce_prompt_split) == 1:
             programs.append(
-                ProgramInfo(int(enforce_prompt_split[0]), 0, ">=", 0, ">=")
+                ProgramInfo(program_id, 0, ">=", 0, ">=")
             )  # this will always satisfy
         else:
-            program_id = enforce_prompt_split[0]
             enforce_batch_size, enforce_prompt_length = (
                 _ for _ in enforce_prompt_split[1].split(",")
             )
@@ -470,7 +470,8 @@ def parse_program_limit(limit_str: str) -> tuple[int, str]:
 
     if len(programs) == 0:
         programs = [
-            ProgramInfo(p.program_id, 0, ">=", 0, ">=") for p in program_criteria_list
+            ProgramInfo(str(p.program_id), 0, ">=", 0, ">=")
+            for p in program_criteria_list
         ]