Merge pull request #209 from ImperialCollegeLondon/fix-parquet-read-s…

…peed-example Fix parquet read speed example
ImperialCollegeLondon · Jan 12, 2025 · 5fe8b78 · 5fe8b78
2 parents 2d02893 + 9a724c7
commit 5fe8b78
Show file tree

Hide file tree

Showing 9 changed files with 86 additions and 10 deletions.
diff --git a/docs/source/examples/analysing-GITT-data.ipynb b/docs/source/examples/analysing-GITT-data.ipynb
@@ -19,7 +19,15 @@
    "outputs": [],
    "source": [
     "%%capture\n",
-    "%pip install matplotlib\n",
+    "%pip install matplotlib"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import matplotlib.pyplot as plt\n",
     "\n",

diff --git a/docs/source/examples/comparing-pyprobe-performance.ipynb b/docs/source/examples/comparing-pyprobe-performance.ipynb
@@ -17,7 +17,15 @@
    "source": [
     "%%capture\n",
     "%pip install matplotlib\n",
-    "%pip install pandas\n",
+    "%pip install pandas"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import pandas as pd\n",
     "import polars as pl\n",
@@ -480,8 +488,10 @@
     "        overwrite_existing=True,\n",
     "        compression_priority=priority,\n",
     "    )\n",
-    "    file_sizes[i] = os.path.getsize(data_directory + f\"/sample_data_neware.parquet\")\n",
-    "    pyprobe_time, _, _ = measure_pyprobe(repeats, \"sample_data_neware.parquet\")\n",
+    "    file_sizes[i] = os.path.getsize(\n",
+    "        data_directory + f\"/sample_data_neware_test.parquet\"\n",
+    "    )\n",
+    "    pyprobe_time, _, _ = measure_pyprobe(repeats, \"sample_data_neware_test.parquet\")\n",
     "    times[i, :] = pyprobe_time[-1, :]\n",
     "\n",
     "file_sizes = np.append(\n",

diff --git a/docs/source/examples/differentiating-voltage-data.ipynb b/docs/source/examples/differentiating-voltage-data.ipynb
@@ -25,7 +25,15 @@
    "outputs": [],
    "source": [
     "%%capture\n",
-    "%pip install matplotlib\n",
+    "%pip install matplotlib"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import matplotlib.pyplot as plt\n",
     "\n",

diff --git a/docs/source/examples/filtering-data.ipynb b/docs/source/examples/filtering-data.ipynb
@@ -22,7 +22,15 @@
    "outputs": [],
    "source": [
     "%%capture\n",
-    "%pip install matplotlib\n",
+    "%pip install matplotlib"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import matplotlib.pyplot as plt\n",
     "\n",

diff --git a/docs/source/examples/getting-started.ipynb b/docs/source/examples/getting-started.ipynb
@@ -14,7 +14,15 @@
    "outputs": [],
    "source": [
     "%%capture\n",
-    "%pip install matplotlib\n",
+    "%pip install matplotlib"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "from pprint import pprint\n",
     "\n",

diff --git a/docs/source/examples/ocv-fitting.ipynb b/docs/source/examples/ocv-fitting.ipynb
@@ -16,7 +16,15 @@
    "outputs": [],
    "source": [
     "%%capture\n",
-    "%pip install matplotlib\n",
+    "%pip install matplotlib"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import polars as pl\n",
     "from pyprobe.analysis import degradation_mode_analysis as dma\n",

diff --git a/docs/source/examples/plotting.ipynb b/docs/source/examples/plotting.ipynb
@@ -22,7 +22,15 @@
    "outputs": [],
    "source": [
     "%%capture\n",
-    "%pip install matplotlib\n",
+    "%pip install matplotlib"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import matplotlib.pyplot as plt\n",
     "\n",

diff --git a/docs/source/examples/sharing-data.ipynb b/docs/source/examples/sharing-data.ipynb
@@ -11,6 +11,16 @@
     "First we will import some sample data:"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%capture\n",
+    "%pip install matplotlib"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

diff --git a/docs/source/examples/working-with-pybamm-models.ipynb b/docs/source/examples/working-with-pybamm-models.ipynb
@@ -27,7 +27,15 @@
     "%%capture\n",
     "%pip install pybamm\n",
     "%pip install matplotlib\n",
-    "%pip install ipywidgets\n",
+    "%pip install ipywidgets"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "import pyprobe\n",
     "import pybamm\n",
     "import matplotlib.pyplot as plt\n",