add

mosaicml · Oct 10, 2023 · 0540bb7 · 0540bb7
1 parent f39d206
commit 0540bb7
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.github/mcp/mcp_pytest.py b/.github/mcp/mcp_pytest.py
@@ -142,6 +142,16 @@
 
     python -m coverage report
     '''
+    config = RunConfig(
+        name=name,
+        cluster=args.cluster,
+        gpu_type=args.gpu_type,
+        gpu_num=args.gpu_num,
+        image=args.image,
+        integrations=[git_integration],
+        command=command + run_command,
+        scheduling={'max_duration': args.timeout / 60 / 60},
+    )
 
     # Create run
     run = create_run(config)