MilesCranmer commited on
Commit
9065874
1 Parent(s): 86d9c0b

Add missing pickle import

Browse files
Files changed (1) hide show
  1. examples/pysr_demo.ipynb +3 -1
examples/pysr_demo.ipynb CHANGED
@@ -1302,7 +1302,7 @@
1302
  "cell_type": "markdown",
1303
  "metadata": {},
1304
  "source": [
1305
- "We can now load the data:"
1306
  ]
1307
  },
1308
  {
@@ -1311,6 +1311,8 @@
1311
  "metadata": {},
1312
  "outputs": [],
1313
  "source": [
 
 
1314
  "nnet_recordings = pkl.load(open(\"nnet_recordings.pkl\", \"rb\"))\n",
1315
  "f_input = nnet_recordings[\"f_input\"]\n",
1316
  "f_output = nnet_recordings[\"f_output\"]\n",
 
1302
  "cell_type": "markdown",
1303
  "metadata": {},
1304
  "source": [
1305
+ "We can now load the data, including after a crash (be sure to re-run the import cells at the top of this notebook, including the one that starts PyJulia)."
1306
  ]
1307
  },
1308
  {
 
1311
  "metadata": {},
1312
  "outputs": [],
1313
  "source": [
1314
+ "import pickle as pkl\n",
1315
+ "\n",
1316
  "nnet_recordings = pkl.load(open(\"nnet_recordings.pkl\", \"rb\"))\n",
1317
  "f_input = nnet_recordings[\"f_input\"]\n",
1318
  "f_output = nnet_recordings[\"f_output\"]\n",