ci-cd update

catalyst-team · Dec 27, 2021 · f81611d · f81611d
1 parent 1d0a0b2
commit f81611d
Showing 1 changed file with 18 additions and 22 deletions.
diff --git a/examples/notebooks/colab_ci_cd.ipynb b/examples/notebooks/colab_ci_cd.ipynb
@@ -5,8 +5,7 @@
     "colab": {
       "name": "Catalyst: colab minimal CI/CD",
       "provenance": [],
-      "collapsed_sections": [],
-      "toc_visible": true
+      "collapsed_sections": []
     },
     "kernelspec": {
       "name": "python3",
@@ -56,27 +55,13 @@
     {
       "cell_type": "code",
       "metadata": {
-        "id": "va2uPF3hC834",
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "outputId": "092bb97b-9d7e-427c-b0b0-cccae3deb65a"
+        "id": "va2uPF3hC834"
       },
       "source": [
-        "! git clone https://github.com/catalyst-team/catalyst"
+        "! export GIT_REPO=\"catalyst-team/catalyst\" && git clone https://github.com/${GIT_REPO}"
       ],
       "execution_count": null,
-      "outputs": [
-        {
-          "output_type": "stream",
-          "text": [
-            "remote: Total 51407 (delta 2966), reused 3691 (delta 2861), pack-reused 47522\u001b[K\n",
-            "Receiving objects: 100% (51407/51407), 41.49 MiB | 3.77 MiB/s, done.\n",
-            "Resolving deltas: 100% (38574/38574), done.\n"
-          ],
-          "name": "stdout"
-        }
-      ]
+      "outputs": []
     },
     {
       "cell_type": "markdown",
@@ -118,7 +103,6 @@
         "! python -c \"req = open('./catalyst/requirements/requirements.txt').read().replace('>', '=') ; open('./catalyst/requirements/requirements.txt', 'w').write(req)\"\n",
         "! python -c \"req = open('./catalyst/requirements/requirements-cv.txt').read().replace('>', '=') ; open('./catalyst/requirements/requirements-cv.txt', 'w').write(req)\"\n",
         "! python -c \"req = open('./catalyst/requirements/requirements-ml.txt').read().replace('>', '=') ; open('./catalyst/requirements/requirements-ml.txt', 'w').write(req)\"\n",
-        "! python -c \"req = open('./catalyst/requirements/requirements-hydra.txt').read().replace('>', '=') ; open('./catalyst/requirements/requirements-hydra.txt', 'w').write(req)\"\n",
         "! python -c \"req = open('./catalyst/requirements/requirements-optuna.txt').read().replace('>', '=') ; open('./catalyst/requirements/requirements-optuna.txt', 'w').write(req)\""
       ],
       "execution_count": null,
@@ -139,7 +123,8 @@
         "  -r ./catalyst/requirements/requirements-ml.txt \\\n",
         "  -r ./catalyst/requirements/requirements-mlflow.txt \\\n",
         "  -r ./catalyst/requirements/requirements-onnx-gpu.txt \\\n",
-        "  -r ./catalyst/requirements/requirements-optuna.txt \\"
+        "  -r ./catalyst/requirements/requirements-optuna.txt \\\n",
+        "  -r ./catalyst/requirements/requirements-profiler.txt \\"
       ],
       "execution_count": null,
       "outputs": []
@@ -232,11 +217,22 @@
         "id": "lwSvMPYhGgMn"
       },
       "source": [
-        "! cd catalyst && pytest ."
+        "! cd catalyst && CATALYST_COMPUTE_PER_CLASS_METRICS=\"1\" OMP_NUM_THREADS=\"1\" MKL_NUM_THREADS=\"1\" pytest ."
       ],
       "execution_count": null,
       "outputs": []
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "! cd catalyst && BENCHMARK_REQUIRED=\"1\" pytest ./tests/benchmarks"
+      ],
+      "metadata": {
+        "id": "-zETFyoOZNV5"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "markdown",
       "metadata": {