speedup influence sentiment analysis notebook in CI

Author: jakobkruse1

notebooks/influence_sentiment_analysis.ipynb

@@ -84,6 +84,7 @@
     }
    ],
    "source": [
+    "import os\n",
     "from copy import deepcopy\n",
     "from typing import Sequence\n",
     "\n",
@@ -130,7 +131,8 @@
    "source": [
     "seed = 42\n",
     "torch.manual_seed(seed)\n",
-    "torch.cuda.manual_seed(seed)"
+    "torch.cuda.manual_seed(seed)\n",
+    "is_CI = os.environ.get(\"CI\")"
    ]
   },
   {
@@ -577,8 +579,8 @@
     }
    ],
    "source": [
-    "NUM_TRAIN_EXAMPLES = 100\n",
-    "NUM_TEST_EXAMPLES = 100\n",
+    "NUM_TRAIN_EXAMPLES = 100 if not is_CI else 7\n",
+    "NUM_TEST_EXAMPLES = 100 if not is_CI else 5\n",
     "\n",
     "small_train_dataset = (\n",
     "    imdb[\"train\"]\n",