flag _cli_desc with comments pointing out that this describes the module

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."
 

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.
 

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.