Fixed documentation build dependencies installation

Author: gperdrizet

.github/workflows/docs.yml

@@ -44,12 +44,13 @@ jobs:
         cache: 'pip'  # Cache pip dependencies for faster builds
 
     # Step 3: Install package with documentation dependencies
-    # The [docs] extra includes: sphinx, sphinx-rtd-theme, nbsphinx,
-    # sphinx-autodoc-typehints, ipykernel, and pandoc
+    # Note: Docs dependencies are in a Poetry group, not extras,
+    # so we install them directly with pip
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -e ".[docs]"
+        pip install sphinx sphinx-rtd-theme nbsphinx sphinx-autodoc-typehints ipykernel pandoc
+        pip install -e .
         
     # Step 4: Build HTML documentation using Sphinx
     # Uses sphinx-build directly (Makefile has been removed for simplicity)

README.md

@@ -1,5 +1,9 @@
 # PyTorch: CIFAR-10 Demonstration
 
+[![Publish to PyPI](https://github.com/gperdrizet/CIFAR10/actions/workflows/publish-to-pypi.yml/badge.svg)](https://github.com/gperdrizet/CIFAR10/actions/workflows/publish-to-pypi.yml)
+
+[![Deploy Documentation](https://github.com/gperdrizet/CIFAR10/actions/workflows/docs.yml/badge.svg)](https://github.com/gperdrizet/CIFAR10/actions/workflows/docs.yml)
+
 A progressive deep learning tutorial for image classification on the CIFAR-10 dataset using PyTorch. This project demonstrates the evolution from basic deep neural networks to optimized convolutional neural networks with data augmentation. It also provides a set of utility functions as a PyPI package for use in other projects.
 
 [View on PyPI](https://pypi.org/project/cifar10_tools) | [Documentation](https://gperdrizet.github.io/CIFAR10/)