Add environment variable check cgpotts#64

dattgoswami · Aug 17, 2020 · 4c9174a · 4c9174a
1 parent 1a58d90
commit 4c9174a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/hw_colors.ipynb b/hw_colors.ipynb
@@ -487,7 +487,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%time _ = dev_mod.fit(dev_cols_train, dev_seqs_train)"
+    "if 'IS_GRADESCOPE_ENV' not in os.environ:\n",
+    "    %time _ = dev_mod.fit(dev_cols_train, dev_seqs_train)\n",
+    "else:\n",
+    "    dev_mod.fit(dev_cols_train, dev_seqs_train)"
    ]
   },
   {