enable unyt 3.0

vbr-calc · Jan 8, 2024 · d82df67 · d82df67
1 parent 24d02d2
commit d82df67
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/pyVBRc/tests/test_pyVBRc.py b/pyVBRc/tests/test_pyVBRc.py
@@ -46,6 +46,12 @@ def test_interpolator(fname):
     dgval = vbr.input.SV.dg_um.min()
     phi_targets = np.full(T_targets.shape, np.log10(phival))
     dg_targets = np.full(T_targets.shape, np.log10(dgval))
+    if isinstance(T_targets, unyt_array):
+        T_targets = T_targets.d
+    if isinstance(dg_targets, unyt_array):
+        dg_targets = dg_targets.d
+    if isinstance(phi_targets, unyt_array):
+        phi_targets = phi_targets.d
     targets = np.column_stack((T_targets, phi_targets, dg_targets))
     Vs_interp = interp(targets)
     assert len(Vs_interp) == nT * 2

diff --git a/pyproject.toml b/pyproject.toml
@@ -24,7 +24,7 @@ dependencies=["numpy",
               "scipy",
               "matplotlib",
               "packaging",
-              "unyt<3.0",
+              "unyt",
  ]
 
 [tool.setuptools]