101 thru papermill

securefederatedai · Jan 16, 2025 · 0186bf3 · 0186bf3
1 parent b8e2c70
commit 0186bf3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/.github/workflows/workflow_interface_101_mnist.yml b/.github/workflows/workflow_interface_101_mnist.yml
@@ -31,16 +31,16 @@ jobs:
       with:
         python-version: "3.10"
 
-    - name: Install Jupyter Lab Package
-      run: pip install jupyterlab
+    - name: Install Papermill Package
+      run: pip install papermill
 
     - name: Run Notebook
       run: |
-        jupyter nbconvert --execute --to notebook ./openfl-tutorials/experimental/workflow/101_MNIST.ipynb
+        papermill ./openfl-tutorials/experimental/workflow/101_MNIST.ipynb ./openfl-tutorials/experimental/workflow/101_MNIST_output.ipynb --request-save-on-cell-execute --log-output --autosave-cell-every 30
         echo "Notebook run completed"
 
     - name: Tar files
-      run: tar -cvf notebook.tar ./openfl-tutorials/experimental/workflow/101_MNIST.nbconvert.ipynb
+      run: tar -cvf notebook.tar ./openfl-tutorials/experimental/workflow/101_MNIST_output.ipynb
 
     - name: Upload Artifacts
       uses: actions/upload-artifact@v4

diff --git a/openfl-tutorials/experimental/workflow/101_MNIST.ipynb b/openfl-tutorials/experimental/workflow/101_MNIST.ipynb
@@ -67,10 +67,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Below code will display the print statement output on screen as well\n",
-    "import sys\n",
-    "sys.stdout = open('/dev/stdout', 'w')\n",
-    "\n",
     "!pip install git+https://github.com/securefederatedai/openfl.git\n",
     "!pip install -r workflow_interface_requirements.txt\n",
     "!pip install torch\n",