add pragma no cover

mllam_data_prep/__init__.py

@@ -2,9 +2,9 @@
 
 import importlib.metadata
 
-try:
+try:  # pragma: no cover
     __version__ = importlib.metadata.version(__name__)
-except importlib.metadata.PackageNotFoundError:
+except importlib.metadata.PackageNotFoundError:  # pragma: no cover
     __version__ = "unknown"
 
 # expose the public API

mllam_data_prep/__main__.py

@@ -1,21 +1,20 @@
-# pragma: no cover
-import os
-from pathlib import Path
+import os  # pragma: no cover
+from pathlib import Path  # pragma: no cover
 
-from loguru import logger
+from loguru import logger  # pragma: no cover
 
-from .create_dataset import create_dataset_zarr
+from .create_dataset import create_dataset_zarr  # pragma: no cover
 
 # Attempt to import psutil and dask.distributed modules
-DASK_DISTRIBUTED_AVAILABLE = True
-try:
+DASK_DISTRIBUTED_AVAILABLE = True  # pragma: no cover
+try:  # pragma: no cover
     import psutil
     from dask.diagnostics import ProgressBar
     from dask.distributed import LocalCluster
-except ImportError or ModuleNotFoundError:
+except ImportError or ModuleNotFoundError:  # pragma: no cover
     DASK_DISTRIBUTED_AVAILABLE = False
 
-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: no cover
     import argparse
 
     parser = argparse.ArgumentParser(

mllam_data_prep/config.py

@@ -39,7 +39,7 @@ def validate_config(config_inputs):
             elif isinstance(input_dataset.variables, dict):  # pragma: no cover
                 variable_vars = input_dataset.variables.keys()
             else:
-                raise TypeError(
+                raise TypeError(  # pragma: no cover
                     f"Expected an instance of list or dict, but got {type(input_dataset.variables)}."
                 )
             derived_variable_vars = input_dataset.derived_variables.keys()