Skip ocean and sea-ice partitions if they exist

This saves a lot of time when rerunning these steps.
MPAS-Dev · Aug 14, 2023 · 5ea9d57 · 5ea9d57
1 parent d09e120
commit 5ea9d57
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/compass/ocean/tests/global_ocean/files_for_e3sm/ocean_graph_partition.py b/compass/ocean/tests/global_ocean/files_for_e3sm/ocean_graph_partition.py
@@ -76,6 +76,9 @@ def run(self):
             if ncores > ncells:
                 raise ValueError('Can\t have more tasks than cells in a '
                                  'partition file.')
+            out_filename = f'mpas-o.graph.info.{creation_date}.part.{ncores}'
+            if os.path.exists(out_filename):
+                continue
             if ncores == 1:
                 args = ['touch', f'mpas-o.graph.info.{creation_date}.part.1']
             else:

diff --git a/compass/ocean/tests/global_ocean/files_for_e3sm/seaice_graph_partition.py b/compass/ocean/tests/global_ocean/files_for_e3sm/seaice_graph_partition.py
@@ -116,7 +116,16 @@ def run(self):
             if ncores > ncells:
                 raise ValueError('Can\t have more tasks than cells in a '
                                  'partition file.')
-            if ncores > 1:
+            if ncores <= 0:
+                raise ValueError(f'Can\t make a partition for {ncores} tasks.')
+            out_filename = \
+                f'mpas-seaice.graph.info.{creation_date}.part.{ncores}'
+            if os.path.exists(out_filename):
+                continue
+
+            if ncores == 1:
+                args = ['touch', f'mpas-o.graph.info.{creation_date}.part.1']
+            else:
                 args.append(f'{ncores}')
 
         check_call(args, logger)