use pkg_resources for PyTorch version checks in notebooks (openai#191)

* use pkg_resources for PyTorch version checks in notebooks * remove version checking in notebook code cell Co-authored-by: Jong Wook Kim <jongwook@nyu.edu>
lhb5883 · Apr 10, 2022 · e58d494 · e58d494
1 parent 3482bb6
commit e58d494
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/notebooks/Interacting_with_CLIP.ipynb b/notebooks/Interacting_with_CLIP.ipynb
@@ -352,10 +352,9 @@
       "source": [
         "import numpy as np\n",
         "import torch\n",
+        "from pkg_resources import packaging\n",
         "\n",
-        "print(\"Torch version:\", torch.__version__)\n",
-        "\n",
-        "assert torch.__version__.split(\".\") >= [\"1\", \"7\", \"1\"], \"PyTorch 1.7.1 or later is required\""
+        "print(\"Torch version:\", torch.__version__)\n"
       ],
       "execution_count": 2,
       "outputs": [

diff --git a/notebooks/Prompt_Engineering_for_ImageNet.ipynb b/notebooks/Prompt_Engineering_for_ImageNet.ipynb
@@ -588,10 +588,9 @@
         "import torch\n",
         "import clip\n",
         "from tqdm.notebook import tqdm\n",
+        "from pkg_resources import packaging\n",
         "\n",
-        "print(\"Torch version:\", torch.__version__)\n",
-        "\n",
-        "assert torch.__version__.split(\".\") >= [\"1\", \"7\", \"1\"], \"PyTorch 1.7.1 or later is required\""
+        "print(\"Torch version:\", torch.__version__)\n"
       ],
       "execution_count": 2,
       "outputs": [
@@ -1105,4 +1104,4 @@
       ]
     }
   ]
-}
+}