diff --git a/src/gz21_ocean_momentum/cli/data.py b/src/gz21_ocean_momentum/cli/data.py index f841fd37..fe385ba7 100755 --- a/src/gz21_ocean_momentum/cli/data.py +++ b/src/gz21_ocean_momentum/cli/data.py @@ -14,6 +14,7 @@ import xarray as xr import dask.multiprocessing +# Description of this module _cli_desc = "GZ21 data step: download CM2.6 dataset, apply coarse graining \ and generate forcings. Saves result to disk in zarr format." diff --git a/src/gz21_ocean_momentum/cli/infer.py b/src/gz21_ocean_momentum/cli/infer.py index e125e05d..5ca78744 100755 --- a/src/gz21_ocean_momentum/cli/infer.py +++ b/src/gz21_ocean_momentum/cli/infer.py @@ -23,6 +23,7 @@ import gz21_ocean_momentum.train.losses as loss_funcs from gz21_ocean_momentum.inference.utils import predict_lazy_cm2_6 +# Description of this module _cli_desc = """ Use a trained GZ21 neural net to predict forcing for input ocean velocity data. diff --git a/src/gz21_ocean_momentum/cli/train.py b/src/gz21_ocean_momentum/cli/train.py index 28ff8784..b14b43d9 100755 --- a/src/gz21_ocean_momentum/cli/train.py +++ b/src/gz21_ocean_momentum/cli/train.py @@ -34,6 +34,7 @@ # TODO ideally temporary but probably not import copy +# Description of this module _cli_desc = """ Train a Pytorch neural net to predict subgrid ocean momentum forcing from ocean surface velocity.