Merge pull request #53 from salesforce/numba-integration

notebook update
salesforce · Sep 22, 2022 · 7296036 · 7296036
2 parents 3b8e328 + 2c96027
commit 7296036
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 23 deletions.
diff --git a/tutorials/tutorial-2.a-warp_drive_sampler.ipynb b/tutorials/tutorial-2.a-warp_drive_sampler.ipynb
@@ -75,7 +75,7 @@
     "- the pip package manager, OR\n",
     "- by cloning the warp_drive package and installing the requirements.\n",
     "\n",
-    "On Colab, we will do the latter."
+    "We will install the latest version of WarpDrive using the pip package manager."
    ]
   },
   {
@@ -84,16 +84,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import sys\n",
-    "\n",
-    "IN_COLAB = \"google.colab\" in sys.modules\n",
-    "\n",
-    "if IN_COLAB:\n",
-    "    ! git clone https://github.com/salesforce/warp-drive.git\n",
-    "    % cd warp-drive\n",
-    "    ! pip install -e .\n",
-    "else:\n",
-    "    ! pip install -U rl_warp_drive"
+    "pip install -U rl_warp_drive"
    ]
   },
   {

diff --git a/tutorials/tutorial-2.b-warp_drive_sampler.ipynb b/tutorials/tutorial-2.b-warp_drive_sampler.ipynb
@@ -75,7 +75,7 @@
     "- the pip package manager, OR\n",
     "- by cloning the warp_drive package and installing the requirements.\n",
     "\n",
-    "On Colab, we will do the latter."
+    "We will install the latest version of WarpDrive using the pip package manager."
    ]
   },
   {
@@ -84,16 +84,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import sys\n",
-    "\n",
-    "IN_COLAB = \"google.colab\" in sys.modules\n",
-    "\n",
-    "if IN_COLAB:\n",
-    "    ! git clone https://github.com/salesforce/warp-drive.git\n",
-    "    % cd warp-drive\n",
-    "    ! pip install -e .\n",
-    "else:\n",
-    "    ! pip install -U rl_warp_drive"
+    "pip install -U rl_warp_drive"
    ]
   },
   {

diff --git a/tutorials/tutorial-5-training_with_warp_drive.ipynb b/tutorials/tutorial-5-training_with_warp_drive.ipynb
@@ -284,7 +284,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Important! Since v1.8, WarpDrive is fully supporting Numba as the backend. \n",
+    "### Important! Since v2.0, WarpDrive is fully supporting Numba as the backend. \n",
     "\n",
     "Before v1.8: Ensure that 'use_cuda' is set to True (in order to run the simulation on the GPU).\n",
     "\n",