adding allocation to the Shaker maker job submit

NHERI-SimCenter · Oct 2, 2024 · 518e974 · 518e974
1 parent f5dcf72
commit 518e974
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/tools/ShakerMaker/ShakerMakersubmitjob.py b/modules/tools/ShakerMaker/ShakerMakersubmitjob.py
@@ -14,6 +14,7 @@
 parser.add_argument('--tapisfolder', type=str, help='folder to upload the files to')
 parser.add_argument('--username', type=str, help='username for DesignSafe')
 parser.add_argument('--password', type=str, help='password for DesignSafe')
+parser.add_argument('--allocation', type=str, help='allocation for DesignSafe')
 
 
 args = parser.parse_args()
@@ -39,6 +40,7 @@
 
 
 systemcodepath = args.tapisfolder
+allocation = f"-A {args.allocation}"
 faultinfo = metadata['faultdata']
 faultfiles = faultinfo['Faultfilenames']
 
@@ -155,7 +157,7 @@
     'archiveOnAppError': False,
     'fileInputArrays': [{'sourceUrls': urls, 'targetDir': '*'}],
     'parameterSet': {
-        'schedulerOptions': [{'arg': '-A DesignSafe-SimCenter'}],
+        'schedulerOptions': [{'arg': allocation}],
         'envVariables': [
             {'key': 'inputFile', 'value': 'ShakerMakermodel.py'},
             {'key': 'numProcessors', 'value': totalcores},