PySDM 3.0.0rc42__tar.gz → 3.0.0rc43__tar.gz

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysdm
-Version: 3.0.0rc42
+Version: 3.0.0rc43
 Summary: Pythonic particle-based (super-droplet) warm-rain/aqueous-chemistry cloud microphysics package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab
 Author-email: "https://github.com/open-atmos/PySDM/graphs/contributors" <sylwester.arabas@agh.edu.pl>
 License-Expression: GPL-3.0-only

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/attributes/physics/terminal_velocity.py

@@ -21,10 +21,10 @@ class TerminalVelocity(DerivedAttribute):
         )
 
         self.approximation_liquid = particulator.formulae.terminal_velocity_class(
-            particulator
+            particulator.backend
         )
         self.approximation_ice = particulator.formulae.terminal_velocity_ice_class(
-            particulator
+            particulator.backend
         )
 
     def recalculate(self):

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/backends/jax.py

@@ -43,8 +43,6 @@ class Jax(
         self.formulae = formulae or Formulae()
         self.formulae_flattened = self.formulae.flatten
 
-        self.default_jit_flags = {"parallel": False}
-
         methods.CollisionsMethods.__init__(self)
         methods.PairMethods.__init__(self)
         methods.IndexMethods.__init__(self)

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/backends/numba.py

@@ -1,5 +1,6 @@
 """
-Multi-threaded CPU backend using LLVM-powered just-in-time compilation
+CPU backend using LLVM-powered just-in-time compilation
+with multi-threading for multi-cell simulations (i.e., if n_dims>0)
 """
 
 import os
@@ -49,13 +50,15 @@ class Numba(  # pylint: disable=too-many-ancestors,duplicate-code
     ):
         if not double_precision:
             raise NotImplementedError()
+
         self.formulae = formulae or Formulae()
         self.formulae_flattened = self.formulae.flatten
         self.n_dims = n_dims
-        parallel_default = True
 
+        parallel_default = n_dims > 0
         if override_jit_flags is not None and "parallel" in override_jit_flags:
             parallel_default = override_jit_flags["parallel"]
+            del override_jit_flags["parallel"]
 
         if parallel_default:
             if platform.machine() == "arm64":

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/dynamics/freezing.py

@@ -4,7 +4,7 @@ time-dependent formulation for immersion freezing
 and homogeneous freezing and thaw
 """
 
-from typing import Optional
+from typing import Literal, Optional
 from PySDM.dynamics.impl import register_dynamic
 
 
@@ -13,9 +13,9 @@ class Freezing:  # pylint: disable=too-many-instance-attributes
     def __init__(
         self,
         *,
-        homogeneous_freezing: Optional[str] = None,
-        immersion_freezing: Optional[str] = None,
-        thaw: Optional[str] = None,
+        homogeneous_freezing: Optional[Literal["threshold", "time-dependent"]] = None,
+        immersion_freezing: Optional[Literal["singular", "time-dependent"]] = None,
+        thaw: Optional[Literal["instantaneous"]] = None,
     ):
         assert (
             homogeneous_freezing or immersion_freezing or thaw

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/dynamics/terminal_velocity/columnar_ice_crystal.py

@@ -7,11 +7,11 @@ Eq. (18) .Assumed shape is columnar based on empirical parameterizations of
 
 
 class ColumnarIceCrystal:  # pylint: disable=too-few-public-methods,too-many-arguments
-    def __init__(self, particulator):
-        self.particulator = particulator
+    def __init__(self, backend):
+        self.backend = backend
 
     def __call__(self, output, signed_water_mass, cell_id, temperature, pressure):
-        self.particulator.backend.terminal_velocity_columnar_ice_crystals(
+        self.backend.terminal_velocity_columnar_ice_crystals(
             values=output.data,
             signed_water_mass=signed_water_mass.data,
             cell_id=cell_id.data,

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/dynamics/terminal_velocity/gunn_and_kinzer.py

@@ -13,8 +13,8 @@ from PySDM.physics import constants as const
 
 
 class GunnKinzer1949:  # pylint: disable=too-few-public-methods
-    def __init__(self, particulator, small_r_limit=None):
-        self.particulator = particulator
+    def __init__(self, backend, small_r_limit=None):
+        self.backend = backend
 
         """
         Table 2 in
@@ -121,9 +121,9 @@ class GunnKinzer1949:  # pylint: disable=too-few-public-methods
         approximation_small = TpDependent.make(only_small=True)
         small_r_limit = small_r_limit or 40 * const.si.um
         approximation_small(u[1:], space[1:], small_r_limit)
-        self.a = particulator.backend.Storage.from_ndarray(u)
+        self.a = backend.Storage.from_ndarray(u)
         b = np.append(np.diff(u), [u[-1] - u[-2]]) / step
-        self.b = particulator.backend.Storage.from_ndarray(b)
+        self.b = backend.Storage.from_ndarray(b)
 
     def __call__(self, output, radius):
         r_max = radius.amax()
@@ -132,7 +132,7 @@ class GunnKinzer1949:  # pylint: disable=too-few-public-methods
                 f"Radii can be interpolated up to {self.maximum_radius} m"
                 + f" (max value of {r_max} m within input data)"
             )
-        self.particulator.backend.gunn_and_kinzer_interpolation(
+        self.backend.gunn_and_kinzer_interpolation(
             output=output, radius=radius, factor=self.factor, b=self.a, c=self.b
         )
 

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/dynamics/terminal_velocity/power_series.py

@@ -16,8 +16,8 @@ from PySDM.physics import si, constants as const
 
 
 class PowerSeries:  # pylint: disable=too-few-public-methods
-    def __init__(self, particulator, *, prefactors=None, powers=None):
-        self.particulator = particulator
+    def __init__(self, backend, *, prefactors=None, powers=None):
+        self.backend = backend
         self.prefactors = np.array(prefactors or [2.0e-1 * si.m / si.s / np.sqrt(si.m)])
         self.powers = np.array(powers or [1 / 6])
         assert len(self.prefactors) == len(self.powers)
@@ -25,7 +25,7 @@ class PowerSeries:  # pylint: disable=too-few-public-methods
             self.prefactors[i] *= const.PI_4_3**p / si.um ** (3 * p)
 
     def __call__(self, output, radius):
-        self.particulator.backend.power_series(
+        self.backend.power_series(
             values=output.data,
             radius=radius.data,
             num_terms=len(self.powers),

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/dynamics/terminal_velocity/rogers_and_yau.py

@@ -5,11 +5,11 @@ equations: (8.5), (8.6), (8.8)
 
 
 class RogersYau:  # pylint: disable=too-few-public-methods
-    def __init__(self, particulator):
-        self.particulator = particulator
+    def __init__(self, backend):
+        self.backend = backend
 
     def __call__(self, output, radius):
-        self.particulator.backend.rogers_and_yau_terminal_velocity(
+        self.backend.rogers_and_yau_terminal_velocity(
             values=output.data,
             radius=radius.data,
         )

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/PySDM/dynamics/terminal_velocity/spheres_ice.py

@@ -4,11 +4,11 @@ terminal velocity of solid ice spheres
 
 
 class IceSphere:  # pylint: disable=too-few-public-methods,too-many-arguments
-    def __init__(self, particulator):
-        self.particulator = particulator
+    def __init__(self, backend):
+        self.backend = backend
 
     def __call__(self, output, signed_water_mass, cell_id, temperature, pressure):
-        self.particulator.backend.terminal_velocity_ice_spheres(
+        self.backend.terminal_velocity_ice_spheres(
             values=output.data,
             signed_water_mass=signed_water_mass.data,
             cell_id=cell_id.data,

pysdm-3.0.0rc43/PySDM/initialisation/init_fall_momenta.py

@@ -0,0 +1,41 @@
+"""
+Initialize the `PySDM.attributes.physics.relative_fall_velocity.RelativeFallMomentum`
+of droplets
+"""
+
+import numpy as np
+
+
+def init_fall_momenta(
+    *,
+    terminal_velocity_approx,
+    backend,
+    water_mass: np.ndarray,
+    zero: bool = False,
+):
+    """
+    Calculate default values of the
+    `PySDM.attributes.physics.relative_fall_velocity.RelativeFallMomentum` attribute
+    (needed when using
+    `PySDM.attributes.physics.relative_fall_velocity.RelativeFallVelocity` attribute)
+
+    Parameters:
+        - water_mass: a numpy array of superdroplet water masses
+
+    Returns:
+        - a numpy array of initial momentum values
+    """
+    if zero:
+        return np.zeros_like(water_mass)
+
+    approximation = terminal_velocity_approx(backend=backend)
+
+    volume_arr = backend.formulae.particle_shape_and_density.mass_to_volume(water_mass)
+    radii_arr = backend.formulae.trivia.radius(volume=volume_arr)
+    radii = backend.Storage.from_ndarray(radii_arr)
+
+    output = backend.Storage.empty((len(water_mass),), dtype=float)
+
+    approximation(output=output, radius=radii)
+
+    return output.to_ndarray() * water_mass

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Abade_and_Albuquerque_2024/fig_2.ipynb

@@ -87,7 +87,6 @@
     "        freezing_temperature_spectrum=\"Niemand_et_al_2012\",\n",
     "        heterogeneous_ice_nucleation_rate=\"ABIFM\",\n",
     "    ),\n",
-    "    override_jit_flags={\"parallel\": False}\n",
     ")"
    ]
   },

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Arabas_and_Shima_2017/simulation.py

@@ -30,14 +30,7 @@ class Simulation:
             initial_water_vapour_mixing_ratio=settings.initial_water_vapour_mixing_ratio,
             T0=settings.T0,
             w=settings.w,
-            backend=backend(
-                formulae=settings.formulae,
-                **(
-                    {"override_jit_flags": {"parallel": False}}
-                    if backend is Numba
-                    else {}
-                ),
-            ),
+            backend=backend(formulae=settings.formulae),
         )
         r_wet = equilibrate_wet_radii(
             r_dry=r_dry,

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Arabas_et_al_2025/aida.ipynb

@@ -206,7 +206,7 @@
     "                T0 = settings.T0,\n",
     "                initial_water_vapour_mixing_ratio = settings.initial_water_vapour_mixing_ratio,\n",
     "                w = vertical_velocity,\n",
-    "                backend=CPU(settings.formulae, override_jit_flags={\"parallel\": False}),\n",
+    "                backend=CPU(settings.formulae),\n",
     "                mixed_phase = True\n",
     "            )\n",
     "        dry_radius = settings.formulae.trivia.radius(volume=settings.attributes['dry volume'])\n",

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Arabas_et_al_2025/make_particulator.py

@@ -35,7 +35,7 @@ def make_particulator(
         "particle_shape_and_density": "MixedPhaseSpheres",
     }
     formulae = Formulae(**formulae_ctor_args)
-    backend = CPU(formulae, override_jit_flags={"parallel": False})
+    backend = CPU(formulae)
 
     attributes = {
         "signed water mass": np.ones(n_sd) * droplet_volume * formulae.constants.rho_w

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Grabowski_and_Pawlowska_2023/simulation.py

@@ -26,9 +26,7 @@ class Simulation(BasicSimulation):
             T0=settings.initial_temperature,
             w=settings.vertical_velocity,
             mass_of_dry_air=44 * si.kg,
-            backend=CPU(
-                formulae=settings.formulae, override_jit_flags={"parallel": False}
-            ),
+            backend=CPU(formulae=settings.formulae),
         )
         volume = environment.mass_of_dry_air / settings.initial_air_density
         attributes = {

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Jensen_and_Nugent_2017/simulation.py

@@ -38,10 +38,7 @@ class Simulation(BasicSimulation):
             T0=settings.T0,
             w=settings.vertical_velocity,
             z0=settings.z0,
-            backend=CPU(
-                formulae=settings.formulae,
-                override_jit_flags={"parallel": False},
-            ),
+            backend=CPU(formulae=settings.formulae),
         )
 
         self.r_dry, n_in_unit_volume = Logarithmic(

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Kreidenweis_et_al_2003/simulation.py

@@ -19,9 +19,7 @@ class Simulation(BasicSimulation):
             initial_water_vapour_mixing_ratio=settings.initial_water_vapour_mixing_ratio,
             T0=settings.T0,
             w=settings.w,
-            backend=CPU(
-                formulae=settings.formulae, override_jit_flags={"parallel": False}
-            ),
+            backend=CPU(formulae=settings.formulae),
         )
         attributes = environment.init_attributes(
             n_in_dv=settings.n_in_dv,

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Lowe_et_al_2019/settings.py

@@ -23,7 +23,6 @@ def backend(model):
             optical_albedo="Bohren1987",
             optical_depth="Stephens1978",
         ),
-        override_jit_flags={"parallel": False},
     )
 
 

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Luettmer_et_al_2026/commons.py

@@ -27,7 +27,6 @@ def run_simulations(setting):
 
 
 def hom_pure_droplet_freezing_backend():
-    cmn = {"override_jit_flags": {"parallel": False}}
     backends = {
         "threshold": CPU(
             formulae=Formulae(
@@ -35,7 +34,6 @@ def hom_pure_droplet_freezing_backend():
                 homogeneous_ice_nucleation_rate="Null",
                 saturation_vapour_pressure="MurphyKoop2005",
             ),
-            **cmn,
         ),
         "KoopMurray2016": CPU(
             formulae=Formulae(
@@ -43,7 +41,6 @@ def hom_pure_droplet_freezing_backend():
                 homogeneous_ice_nucleation_rate="KoopMurray2016",
                 saturation_vapour_pressure="MurphyKoop2005",
             ),
-            **cmn,
         ),
         "Spichtinger2023": CPU(
             formulae=Formulae(
@@ -51,7 +48,6 @@ def hom_pure_droplet_freezing_backend():
                 homogeneous_ice_nucleation_rate="Koop_Correction",
                 saturation_vapour_pressure="MurphyKoop2005",
             ),
-            **cmn,
         ),
         "Koop2000": CPU(
             formulae=Formulae(
@@ -59,7 +55,6 @@ def hom_pure_droplet_freezing_backend():
                 homogeneous_ice_nucleation_rate="Koop2000",
                 saturation_vapour_pressure="MurphyKoop2005",
             ),
-            **cmn,
         ),
         "KoopMurray2016_DWA": CPU(
             formulae=Formulae(
@@ -67,7 +62,6 @@ def hom_pure_droplet_freezing_backend():
                 homogeneous_ice_nucleation_rate="KoopMurray2016_DWA",
                 saturation_vapour_pressure="MurphyKoop2005",
             ),
-            **cmn,
         ),
     }
     return backends

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Niedermeier_et_al_2014/simulation.py

@@ -24,10 +24,7 @@ class Simulation(BasicSimulation):
             mass_of_dry_air=settings.mass_of_dry_air,
             w=settings.vertical_velocity,
             mixed_phase=True,
-            backend=CPU(
-                settings.formulae,
-                override_jit_flags={"parallel": False},
-            ),
+            backend=CPU(settings.formulae),
         )
 
         air_volume = settings.mass_of_dry_air / settings.rhod0

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Pyrcel/simulation.py

@@ -30,9 +30,7 @@ class Simulation(BasicSimulation):
             T0=settings.initial_temperature,
             w=settings.vertical_velocity,
             mass_of_dry_air=mass_of_dry_air,
-            backend=CPU(
-                formulae=settings.formulae, override_jit_flags={"parallel": False}
-            ),
+            backend=CPU(formulae=settings.formulae),
         )
         n_sd = sum(settings.n_sd_per_mode)
         volume = environment.mass_of_dry_air / settings.initial_air_density

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Rozanski_and_Sonntag_1982/figs_4_5_6.ipynb

@@ -92,7 +92,7 @@
     "    isotope_meteoric_water_line='Dansgaard1964',\n",
     "    isotope_ratio_evolution='MerlivatAndJouzel1979',\n",
     ")\n",
-    "BACKEND = CPU(FORMULAE, override_jit_flags={'parallel': False})\n",
+    "BACKEND = CPU(FORMULAE)\n",
     "CONST = FORMULAE.constants\n",
     "TRIVIA = FORMULAE.trivia\n",
     "\n",

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Spichtinger_et_al_2023/simulation.py

@@ -31,14 +31,7 @@ class Simulation(BasicSimulation):
             initial_water_vapour_mixing_ratio=settings.initial_water_vapour_mixing_ratio,
             T0=settings.initial_temperature,
             w=settings.w_updraft,
-            backend=backend(
-                formulae=settings.formulae,
-                **(
-                    {"override_jit_flags": {"parallel": False}}
-                    if backend is Numba
-                    else {}
-                ),
-            ),
+            backend=backend(formulae=settings.formulae),
         )
 
         self.n_sd = settings.n_sd

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/Yang_et_al_2018/simulation.py

@@ -30,9 +30,7 @@ class Simulation:
             T0=settings.T0,
             w=settings.w,
             z0=settings.z0,
-            backend=backend(
-                formulae=self.formulae, override_jit_flags={"parallel": False}
-            ),
+            backend=backend(formulae=self.formulae),
         )
 
         condensation = Condensation(

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/__init__.py

@@ -11,4 +11,4 @@ except PackageNotFoundError:
     # package is not installed
     pass
 
-# assert PySDM.__version__ == __version__
+assert PySDM.__version__ == __version__

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/examples/PySDM_examples/seeding/simulation.py

@@ -21,9 +21,7 @@ class Simulation:
             initial_water_vapour_mixing_ratio=settings.initial_water_vapour_mixing_ratio,
             p0=settings.initial_total_pressure,
             T0=settings.initial_temperature,
-            backend=CPU(
-                formulae=settings.formulae, override_jit_flags={"parallel": False}
-            ),
+            backend=CPU(formulae=settings.formulae),
         )
         r_dry, n_in_dv = ConstantMultiplicity(
             settings.initial_aerosol_dry_radii

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/pysdm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysdm
-Version: 3.0.0rc42
+Version: 3.0.0rc43
 Summary: Pythonic particle-based (super-droplet) warm-rain/aqueous-chemistry cloud microphysics package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab
 Author-email: "https://github.com/open-atmos/PySDM/graphs/contributors" <sylwester.arabas@agh.edu.pl>
 License-Expression: GPL-3.0-only

pysdm-3.0.0rc43/pysdm.egg-info/scm_version.json

@@ -0,0 +1,8 @@
+{
+  "tag": "3.0.0rc43",
+  "distance": 0,
+  "node": "g62da57648ce2858b6ce98578e3d0234fa133e28f",
+  "dirty": false,
+  "branch": "HEAD",
+  "node_date": "2026-10-12"
+}

pysdm-3.0.0rc43/tests/unit_tests/backends/test_ctor_defaults_and_warnings.py

@@ -0,0 +1,75 @@
+# pylint: disable=missing-module-docstring,missing-class-docstring,missing-function-docstring
+from contextlib import nullcontext
+from unittest import mock
+import platform
+import warnings
+import inspect
+
+import pytest
+import numba
+
+from PySDM.backends import Numba, ThrustRTC
+
+
+class TestCtorDefaultsAndWarnings:
+    @staticmethod
+    def test_gpu_ctor_defaults():
+        signature = inspect.signature(ThrustRTC.__init__)
+        assert signature.parameters["verbose"].default is False
+        assert signature.parameters["debug"].default is False
+        assert signature.parameters["double_precision"].default is False
+        assert signature.parameters["formulae"].default is None
+        assert signature.parameters["n_dims"].default == 0
+
+    @staticmethod
+    def test_cpu_ctor_defaults():
+        signature = inspect.signature(Numba.__init__)
+        assert signature.parameters["formulae"].default is None
+        assert signature.parameters["n_dims"].default == 0
+
+    @staticmethod
+    def test_jax_ctor_defaults():
+        signature = inspect.signature(Numba.__init__)
+        assert signature.parameters["formulae"].default is None
+        assert signature.parameters["n_dims"].default == 0
+
+    @staticmethod
+    @mock.patch("PySDM.backends.numba.prange", new=range)
+    def test_check_numba_threading_warning():
+        if numba.config.DISABLE_JIT:  # pylint: disable=no-member
+            pytest.skip()
+
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            with pytest.raises(ValueError) as exc_info:
+                Numba(n_dims=1)
+            assert exc_info.match(r"^Numba threading enabled but does not work")
+
+    @staticmethod
+    @pytest.mark.parametrize(
+        "setting, outcome, context",
+        (
+            (True, platform.machine() != "arm64", nullcontext()),
+            (False, False, nullcontext()),
+            (
+                True,
+                False,
+                mock.patch(
+                    "numba.parfors.parfor.ensure_parallel_support",
+                    side_effect=numba.core.errors.UnsupportedParforsError("Mock"),
+                ),
+            ),
+            (
+                False,
+                False,
+                mock.patch(
+                    "numba.parfors.parfor.ensure_parallel_support",
+                    side_effect=numba.core.errors.UnsupportedParforsError("Mock"),
+                ),
+            ),
+        ),
+    )
+    def test_numba_override_jit_flags_parallel(setting, outcome, context):
+        with context:
+            backend = Numba(override_jit_flags={"parallel": setting})
+        assert backend.default_jit_flags["parallel"] == outcome

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/tests/unit_tests/dynamics/test_terminal_velocity.py

@@ -34,10 +34,10 @@ def test_approximation(backend_class, plot=False):
         / 100
     )
     u_term_ry = particulator.backend.Storage.empty((len(u),), float)
-    RogersYau(particulator=particulator)(u_term_ry, r)
+    RogersYau(backend=particulator.backend)(u_term_ry, r)
 
     u_term_inter = particulator.backend.Storage.from_ndarray(u_term_ry.to_ndarray())
-    GunnKinzer1949(particulator)(u_term_inter, r)
+    GunnKinzer1949(backend=particulator.backend)(u_term_inter, r)
 
     assert np.mean((u - u_term_ry) ** 2) < 2e-2
     assert np.mean((u - u_term_inter) ** 2) < 1e-6
@@ -107,7 +107,7 @@ def test_power_series(backend_class, prefactors, powers):
     r = particulator.backend.Storage.from_ndarray(r)
 
     u_term_ps = particulator.backend.Storage.empty((len(u),), float)
-    PowerSeries(particulator=particulator, prefactors=prefactors, powers=powers)(
+    PowerSeries(backend=particulator.backend, prefactors=prefactors, powers=powers)(
         u_term_ps, r
     )
 

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/tests/unit_tests/dynamics/test_vapour_deposition_on_ice.py

@@ -76,7 +76,6 @@ def make_particulator(
             dt=dt,
             dv=1 * si.m**3,
             backend=CPU(
-                override_jit_flags={"parallel": False},
                 formulae=Formulae(
                     particle_shape_and_density="MixedPhaseSpheres",
                     diffusion_coordinate=diffusion_coordinate,

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/tests/unit_tests/initialisation/test_init_fall_momenta.py

@@ -10,6 +10,7 @@ from PySDM import Particulator
 from PySDM.environments.box import Box
 from PySDM.initialisation import init_fall_momenta
 from PySDM.physics import si
+from PySDM.dynamics.terminal_velocity import GunnKinzer1949
 
 
 @pytest.fixture(
@@ -57,14 +58,26 @@ class TestInitFallMomenta:
             * params["water mass"]
         )
 
-        assert np.allclose(init_fall_momenta(params["water mass"]), terminal_momentum)
+        assert np.allclose(
+            init_fall_momenta(
+                water_mass=params["water mass"],
+                backend=backend_instance,
+                terminal_velocity_approx=GunnKinzer1949,
+            ),
+            terminal_momentum,
+        )
 
     @staticmethod
-    def test_init_to_zero(params):
+    def test_init_to_zero(params, backend_instance):
         """
         Fall momenta correctly initialized to zero.
         """
 
-        fall_momenta = init_fall_momenta(params["water mass"], zero=True)
+        fall_momenta = init_fall_momenta(
+            water_mass=params["water mass"],
+            zero=True,
+            terminal_velocity_approx=GunnKinzer1949,
+            backend=backend_instance,
+        )
 
         assert (fall_momenta == np.zeros_like(fall_momenta)).all()

{pysdm-3.0.0rc42 → pysdm-3.0.0rc43}/tests/unit_tests/products/test_activation_criteria.py

@@ -23,7 +23,7 @@ from PySDM.initialisation.sampling import spectral_sampling
     "backend",
     (
         pytest.param(GPU(), marks=pytest.mark.xfail(strict=True)),
-        CPU(override_jit_flags={"parallel": False}),
+        CPU(),
     ),
 )
 def test_activation_criteria(backend, plot=False):