Adding gres handling for teach

Adding srun's gres parameter handling logic for teach in order to pass the requested gres and set the default num_gpus if not requested when c=teach and p=gpu

apps/crc_interactive.py

@@ -56,7 +56,8 @@ def __init__(self) -> None:
         cluster_args = self.add_argument_group('Cluster Arguments')
         cluster_args.add_argument('-p', '--partition', help='run the session on a specific partition')
         for cluster, abbrev in self.clusters.items():
-            cluster_args.add_argument(f'-{abbrev}', f'--{cluster}', action='store_true', help=f'launch a session on the {cluster} cluster')
+            cluster_args.add_argument(f'-{abbrev}', f'--{cluster}', action='store_true',
+                                      help=f'launch a session on the {cluster} cluster')
 
         # Arguments for requesting additional hardware resources
         resource_args = self.add_argument_group('Arguments for Increased Resources')
@@ -116,7 +117,7 @@ def parse_args(self, args=None, namespace=None) -> Namespace:
 
         # Set defaults that need to be determined dynamically
         if not args.num_gpus:
-            args.num_gpus = 1 if args.gpu else 0
+            args.num_gpus = 1 if (args.gpu or (args.teach and (args.partition == 'gpu'))) else 0
 
         # Check wall time is between limits, enable both %H:%M format and integer hours
         check_time = args.time.hour + args.time.minute / 60 + args.time.second / 3600
@@ -171,7 +172,7 @@ def create_srun_command(self, args: Namespace) -> str:
                 srun_args += ' ' + srun_arg_name.format(arg_value)
 
         # The --gres argument in srun needs some special handling so is missing from the above dict
-        if (args.gpu or args.invest) and args.num_gpus:
+        if (args.gpu or args.invest or (args.teach and (args.partition == 'gpu'))) and args.num_gpus:
             srun_args += ' ' + f'--gres=gpu:{args.num_gpus}'
 
         try: