Correctly call cuda implementation

smokestacklightnin · Aug 14, 2023 · 2a2b28c · 2a2b28c
1 parent 8caacc5
commit 2a2b28c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/stardis/opacities/tests/whichlibrary.ipynb b/stardis/opacities/tests/whichlibrary.ipynb
@@ -39,7 +39,8 @@
     "            x,\n",
     "            y / np.sqrt(2),\n",
     "            z * (4 * np.pi),\n",
-    "        ).get()\n",
+    "            ret_np_ndarray=True,\n",
+    "        )\n",
     "\n",
     "def call_scipy(x, y, z):\n",
     "    # x, sigma, gamma\n",
@@ -93,8 +94,8 @@
     "x_grid = np.array(range(array_length)) - 0.5 * array_length\n",
     "test_values = (\n",
     "    x_grid,\n",
-    "    np.array(array_length * [4], dtype=float),\n",
-    "    np.array(array_length * [7], dtype=float),\n",
+    "    np.array(array_length * [4]),\n",
+    "    np.array(array_length * [7]),\n",
     ")"
    ]
   },
@@ -117,7 +118,7 @@
     {
      "data": {
       "text/plain": [
-       "<matplotlib.legend.Legend at 0x7f3f94c62a00>"
+       "<matplotlib.legend.Legend at 0x7f141319fb50>"
       ]
      },
      "execution_count": 5,
@@ -151,7 +152,7 @@
     {
      "data": {
       "text/plain": [
-       "[<matplotlib.lines.Line2D at 0x7f3f94b378b0>]"
+       "[<matplotlib.lines.Line2D at 0x7f141313d880>]"
       ]
      },
      "execution_count": 6,