Update mpi4py and pymetis groups to make them optional

.pre-commit-config.yaml

@@ -103,7 +103,7 @@ repos:
       - <<: *uv-managed-hook
         id: mypy
         name: mypy static type checker
-        entry: uv run --group dev --frozen --isolated mypy
+        entry: uv run --group typing-distributed --frozen --isolated mypy
         types_or: [python, pyi]
         pass_filenames: false
 

model/common/pyproject.toml

@@ -43,7 +43,11 @@ version = "0.0.6"
 all = ["icon4py-common[distributed,io]"]
 cuda11 = ['cupy-cuda11x>=13.0', 'gt4py[cuda11]']
 cuda12 = ['cupy-cuda12x>=13.0', 'gt4py[cuda12]']
-distributed = ["ghex>=0.5.0", "mpi4py>=3.1.5"]
+distributed = [
+  "ghex>=0.5.0",
+  "mpi4py>=3.1.5",
+  "pymetis>2022.1"
+]
 io = [
   # external dependencies
   "cartopy>=0.22.0",
@@ -53,7 +57,6 @@ io = [
   "netcdf4>=1.6.1",
   "numpy>=1.23.3",
   "scikit-learn>=1.4.0",
-  "pymetis>2022.1",
   # TODO(halungge): there are failing tests starting from uxarray==2024.4.0: when a data file does not have
   # fields of a given dimension (eg 'edge') then something in uxarray goes wrong with the dimension
   # mapping. It is not yet clear whether this is a uxarray bug or on our side.

pyproject.toml

@@ -55,9 +55,12 @@ typing = [
   "mypy[faster-cache]>=1.13.0",
   "typing-extensions>=4.11.0",
   "types-cffi>=1.16.0",
-  "mpi4py>=4.0.0",
   "scipy-stubs>=1.15.3.0"
 ]
+typing-distributed = [
+  "mpi4py>=3.1.5",
+  "pymetis>=2022.1"
+]
 
 # -- Standard project description options (PEP 621) --
 [project]