Merge pull request #668 from Alejandro-Casanova/patch-25

fix wrongly triggering assert
mrdbourke · Oct 5, 2023 · bd84150 · bd84150
2 parents c87474b + 80c250c
commit bd84150
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extras/exercises/09_pytorch_model_deployment_exercises.ipynb b/extras/exercises/09_pytorch_model_deployment_exercises.ipynb
@@ -71,7 +71,7 @@
     "try:\n",
     "    import torch\n",
     "    import torchvision\n",
-    "    assert int(torch.__version__.split(\".\")[1]) >= 12, \"torch version should be 1.12+\"\n",
+    "    assert int(torch.__version__.split(\".\")[1]) >= 12 or int(torch.__version__.split(\".\")[0]) > 1, \"torch version should be 1.12+\"\n",
     "    assert int(torchvision.__version__.split(\".\")[1]) >= 13, \"torchvision version should be 0.13+\"\n",
     "    print(f\"torch version: {torch.__version__}\")\n",
     "    print(f\"torchvision version: {torchvision.__version__}\")\n",