add constraint for night test

Author: weiyuan-jiang

GEOSldas_App/ldas_setup

@@ -63,7 +63,7 @@ class LDASsetup:
         # ------
         # Optional resource manager input fields
         # ------
-        optSlurmInpKeys = ['job_name', 'qos', 'oserver_nodes', 'writers-per-node']
+        optSlurmInpKeys = ['job_name', 'qos', 'oserver_nodes', 'writers-per-node', 'constraint']
 
 
         # ===============================================================================================
@@ -1562,11 +1562,11 @@ class LDASsetup:
                         '.'.join([expid, 'ldas_err', myDateTime, 'txt']),
                         ]),
                 self.rundir)
-        constraint='cas'
+        constraint = '"[mil|cas]"'
         if self.GEOS_SITE == "NAS":
           constraint = 'cas_ait'
-        elif self.GEOS_SITE == "NCCS":
-          constraint = '"[mil|cas]"'
+        elif 'constraint' in self.optRmInp:
+          constraint = self.optRmInp['constraint']
 
         SBATCHQSUB = 'sbatch'
         if self.GEOS_SITE == 'NAS':
@@ -1852,6 +1852,7 @@ def _printRmInputKeys(rqdRmInpKeys, optRmInpKeys):
     print ('# - writers-per-node = tasks per oserver_node for writing ( default is 5, for future use ),')
     print ('#      IMPORTANT REQUIREMENT: total #writers = writers-per-node * oserver_nodes >= 2')
     print ('#      Jobs will hang when oserver_nodes = writers-per-node = 1.')
+    print ('# - constraint       = mil ; submit the job to a particular machine, defaule is "[mil|cas]")
     print ('#')
     for key in optRmInpKeys:
         print ('#'+key + ':')