Bulk generator: Specify a path to the PAT instead of the PAT itself.

MathiasVP · MathiasVP · commit 7c2612a6a10e · 2025-05-30T12:47:07.000+01:00
diff --git a/misc/scripts/models-as-data/bulk_generate_mad.py b/misc/scripts/models-as-data/bulk_generate_mad.py
@@ -441,11 +441,18 @@ def main(config, args) -> None:
             if experiment_name is None:
                 print("ERROR: --dca argument is required for DCA strategy")
                 sys.exit(1)
-            pat = args.pat
-            if pat is None:
+
+            if args.pat is None:
                 print("ERROR: --pat argument is required for DCA strategy")
                 sys.exit(1)
-            database_results = download_dca_databases(experiment_name, pat, projects)
+            if not os.path.exists(args.pat):
+                print(f"ERROR: Personal Access Token file '{pat}' does not exist.")
+                sys.exit(1)
+            with open(args.pat, "r") as f:
+                pat = f.read().strip()
+                database_results = download_dca_databases(
+                    experiment_name, pat, projects
+                )
 
     # Generate models for all projects
     print("\n=== Generating models ===")
@@ -485,7 +492,7 @@ def main(config, args) -> None:
     parser.add_argument(
         "--pat",
         type=str,
-        help="PAT token to grab DCA databases (the same as the one you use for DCA)",
+        help="Path to a file containing the PAT token required to grab DCA databases (the same as the one you use for DCA)",
         required=False,
     )
     args = parser.parse_args()
