Ensure datapipes is checked for optional imports too.

[coreyjadams]

---

• main
  • Ensure datapipes is checked for optional imports too. #1390 👈
    • Make zarr, xarray, and h5py optional dependencies. #1391

^{Stack generated by Git Town}

[greptile-apps]

Greptile Overview

Greptile Summary

Migrated all datapipes module imports to use OptionalImport pattern for better dependency management.

Key changes:

• Removed datapipes exclusion from .importlinter contract - datapipes now properly checks optional imports
• Replaced hardcoded try/except blocks and ImportError raises with lazy OptionalImport from physicsnemo.core.version_check
• Added TYPE_CHECKING guards where needed for type annotations (e.g., PyGData, netcdf_file)
• Updated all type hints to use string quotes for forward references (e.g., "dali.Pipeline", "xr.Dataset")
• Removed dask.diagnostics.ProgressBar usage in healpix module (simplified to direct compute call)

Packages migrated to OptionalImport:

• nvidia.dali and nvidia.dali.plugin.pytorch
• netCDF4 and scipy.io
• torch_geometric, torch_geometric.data, torch_geometric.loader
• scipy.sparse and scipy.spatial
• sparse_dot_mkl
• tfrecord.torch.dataset
• vtk
• xarray

The changes ensure consistent error messages with install hints across the codebase and defer imports until actual usage.

Important Files Changed

Filename	Overview
.importlinter	Removed datapipes exclusion from import linting - datapipes now uses OptionalImport for optional dependencies
physicsnemo/datapipes/climate/climate.py	Converted DALI, netCDF4, and scipy imports to OptionalImport pattern with proper TYPE_CHECKING guards
physicsnemo/datapipes/gnn/bsms.py	Converted scipy.sparse and sparse_dot_mkl imports to OptionalImport pattern, removed try/except with warning
physicsnemo/datapipes/healpix/data_modules.py	Converted xarray import to OptionalImport pattern, removed dask.diagnostics.ProgressBar usage

[greptile-apps]

_{4 files reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}

[coreyjadams]

/blossom-ci