diff --git a/compass/ocean/tests/isomip_plus/initial_state.py b/compass/ocean/tests/isomip_plus/initial_state.py index c5db82743..09eb2921a 100644 --- a/compass/ocean/tests/isomip_plus/initial_state.py +++ b/compass/ocean/tests/isomip_plus/initial_state.py @@ -40,7 +40,7 @@ class InitialState(Step): Whether the run includes a thin film below grounded ice """ def __init__(self, test_case, resolution, experiment, vertical_coordinate, - time_varying_forcing, thin_film_present): + time_varying_forcing, thin_film_present, include_viz=False): """ Create the step @@ -65,6 +65,7 @@ def __init__(self, test_case, resolution, experiment, vertical_coordinate, Whether the run includes a thin film below grounded ice """ super().__init__(test_case=test_case, name='initial_state') + self.include_viz = include_viz self.resolution = resolution self.experiment = experiment self.vertical_coordinate = vertical_coordinate @@ -88,7 +89,8 @@ def run(self): """ ds, frac = self._compute_initial_condition() self._compute_restoring(ds, frac) - self._plot(ds) + if self.include_viz: + self._plot(ds) def _compute_initial_condition(self): config = self.config diff --git a/compass/ocean/tests/isomip_plus/isomip_plus_test.py b/compass/ocean/tests/isomip_plus/isomip_plus_test.py index cf17731a0..b70181666 100644 --- a/compass/ocean/tests/isomip_plus/isomip_plus_test.py +++ b/compass/ocean/tests/isomip_plus/isomip_plus_test.py @@ -140,7 +140,8 @@ def __init__(self, test_group, resolution, experiment, experiment=experiment, vertical_coordinate=vertical_coordinate, time_varying_forcing=time_varying_forcing, - thin_film_present=thin_film_present)) + thin_film_present=thin_film_present, + include_viz=thin_film_present)) self.add_step( SshAdjustment(test_case=self, resolution=resolution, vertical_coordinate=vertical_coordinate,