Pin pySHiELD to NDSL version 2026.01.00

examples/notebook/test_physics.ipynb

@@ -122,6 +122,7 @@
     "    layout=layout,\n",
     "    tile_partitioner=partitioner.tile,\n",
     "    tile_rank=cs_communicator.tile.rank,\n",
+    "    backend=backend,\n",
     ")\n",
     "\n",
     "# useful for easily allocating distributed data storages (fields)\n",

examples/notebook/test_rad.ipynb

@@ -88,6 +88,7 @@
    "outputs": [],
    "source": [
     "rank = 0\n",
+    "backend = \"numpy\"\n",
     "\n",
     "comm = NullComm(rank, 1)\n",
     "communicator = TileCommunicator.from_layout(comm=comm, layout=(1,1))\n",
@@ -101,8 +102,9 @@
     "    layout=(1,1),\n",
     "    tile_partitioner=communicator.partitioner.tile,\n",
     "    tile_rank=communicator.tile.rank,\n",
+    "    backend=backend\n",
     ")\n",
-    "quantity_factory = QuantityFactory(sizer, backend=\"numpy\")"
+    "quantity_factory = QuantityFactory(sizer, backend=backend)"
    ]
   },
   {

examples/notebook/test_raddriver.ipynb

@@ -74,6 +74,7 @@
    "outputs": [],
    "source": [
     "rank = 0\n",
+    "backend = \"numpy\"\n",
     "\n",
     "comm = NullComm(rank, 1)\n",
     "communicator = TileCommunicator.from_layout(comm=comm, layout=(1,1))\n",
@@ -87,8 +88,9 @@
     "    layout=(1,1),\n",
     "    tile_partitioner=communicator.partitioner.tile,\n",
     "    tile_rank=communicator.tile.rank,\n",
+    "    backend=backend\n",
     ")\n",
-    "quantity_factory = QuantityFactory(sizer, backend=\"numpy\")"
+    "quantity_factory = QuantityFactory(sizer, backend=backend)"
    ]
   },
   {

pyproject.toml

@@ -40,7 +40,7 @@ extras = [
   "pyshield[pyfv3]",
   "pyshield[dev]"
 ]
-ndsl = ["ndsl @ git+https://github.com/NOAA-GFDL/NDSL.git@develop"]
+ndsl = ["ndsl @ git+https://github.com/NOAA-GFDL/NDSL.git@2026.01.00"]
 pyfv3 = ["pyfv3 @ git+https://github.com/NOAA-GFDL/PyFV3.git@develop"]
 test = [
   "pytest",

pyshield/stencils/gfdl_cld_microphysics/gfdl_cld_mp_driver.py

@@ -899,7 +899,7 @@ def make_quantity2d(**kwargs):
         self._convert_mm_day = 86400.0 * constants.RGRAV / self.config.dt_split
 
         self._copy_stencil = stencil_factory.from_origin_domain(
-            basic.copy_defn,
+            basic.copy,
             origin=self._idx.origin_compute(),
             domain=self._idx.domain_compute(),
         )

pyshield/stencils/gfdl_cld_microphysics/terminal_fall.py

@@ -7,7 +7,7 @@
 from ndsl.constants import X_DIM, Y_DIM, Z_DIM
 from ndsl.dsl.gt4py import BACKWARD, FORWARD, PARALLEL, computation, interval
 from ndsl.dsl.typing import FloatField, FloatFieldIJ, IntFieldIJ
-from ndsl.stencils.basic_operations import copy_defn
+from ndsl.stencils.basic_operations import copy
 from pyfv3.stencils.remap_profile import RemapProfile
 from pyshield.stencils.gfdl_cld_microphysics._config import GFDLCloudMPConfig
 
@@ -407,7 +407,7 @@ def __init__(
         # compile stencils
 
         self._copy_stencil = stencil_factory.from_dims_halo(
-            copy_defn,
+            copy,
             compute_dims=dims,
         )
 

pyshield/stencils/physics.py

@@ -21,7 +21,7 @@
 )
 from ndsl.grid import GridData
 from ndsl.logging import ndsl_log
-from ndsl.stencils.basic_operations import copy_defn
+from ndsl.stencils.basic_operations import copy
 from pyshield._config import (
     PHYSICS_PACKAGES,
     TRACER_DIM,
@@ -1242,7 +1242,7 @@ def make_quantity_2d():
         self._sfcvisdfd = make_quantity_2d()
 
         self._copy_stencil = stencil_factory.from_origin_domain(
-            func=copy_defn,
+            func=copy,
             origin=grid_indexing.origin_full(),
             domain=grid_indexing.domain_full(add=(0, 0, 1)),
         )

tests/integration/test_sfc.py

@@ -112,8 +112,8 @@ def states_from_fortran_restarts(
 
 def setup_infrastructure(nx: Int, ny: Int, nz: Int, nzsoil: Int, etafile: Path):
     n_halo = 3
-
     rank = 0
+    backend = "numpy"
 
     comm = NullComm(rank, 1)
     communicator = TileCommunicator.from_layout(comm=comm, layout=(1, 1))
@@ -127,8 +127,9 @@ def setup_infrastructure(nx: Int, ny: Int, nz: Int, nzsoil: Int, etafile: Path):
         layout=(1, 1),
         tile_partitioner=communicator.partitioner.tile,
         tile_rank=communicator.tile.rank,
+        backend=backend,
     )
-    quantity_factory = QuantityFactory(sizer, backend="numpy")
+    quantity_factory = QuantityFactory(sizer, backend=backend)
 
     soil_sizer = SubtileGridSizer.from_tile_params(
         nx_tile=nx,
@@ -139,6 +140,7 @@ def setup_infrastructure(nx: Int, ny: Int, nz: Int, nzsoil: Int, etafile: Path):
         layout=(1, 1),
         tile_partitioner=communicator.partitioner.tile,
         tile_rank=communicator.tile.rank,
+        backend=backend,
     )
     qf_soil = QuantityFactory(soil_sizer, backend="numpy")
 

tests/savepoint/translate/translate_atmos_phy_statein.py

@@ -58,6 +58,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_cloud_frac.py

@@ -344,6 +344,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_cumulative_shalconv.py

@@ -3403,6 +3403,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -3516,6 +3517,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -3621,6 +3623,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -3738,6 +3741,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -3851,6 +3855,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -4018,6 +4023,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -4183,6 +4189,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -4318,6 +4325,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -4379,6 +4387,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -4533,6 +4542,7 @@ def __init__(self, grid, config, stencil_factory):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_final_mp.py

@@ -1034,6 +1034,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -1491,6 +1492,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_gfdl_cld_microphysics.py

@@ -328,6 +328,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_gfs_physics_driver.py

@@ -127,6 +127,7 @@ def compute(self, inputs):
             nz=self.config.npz,
             n_halo=3,
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         quantity_factory = QuantityFactory(sizer, backend=self.stencil_factory.backend)

tests/savepoint/translate/translate_mfpblt.py

@@ -62,6 +62,7 @@ def compute(self, inputs):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         quantity_factory = QuantityFactory(sizer, backend=self.stencil_factory.backend)

tests/savepoint/translate/translate_mfscu.py

@@ -65,6 +65,7 @@ def compute(self, inputs):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         quantity_factory = QuantityFactory(sizer, backend=self.stencil_factory.backend)

tests/savepoint/translate/translate_microphysics.py

@@ -79,6 +79,7 @@ def compute(self, inputs):
             nz=self.config.npz,
             n_halo=3,
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         quantity_factory = QuantityFactory(sizer, backend=self.stencil_factory.backend)

tests/savepoint/translate/translate_mp_full.py

@@ -466,6 +466,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(
@@ -695,6 +696,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_particle_properties.py

@@ -500,6 +500,7 @@ def __init__(
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         self.quantity_factory = QuantityFactory(

tests/savepoint/translate/translate_pbl.py

@@ -97,6 +97,7 @@ def compute(self, inputs):
             n_halo=3,
             data_dimensions={},
             layout=self.config.layout,
+            backend=self.stencil_factory.backend,
         )
 
         quantity_factory = QuantityFactory(sizer, backend=self.stencil_factory.backend)