Merge branch 'main' into master

.github/workflows/test.yml

@@ -17,7 +17,10 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install torch torchvision pytest
+
+        pip install --index-url https://download.pytorch.org/whl/cpu torch torchvision --no-cache-dir
+        pip install pytest
+
         
     - name: Train model
       run: |
@@ -30,7 +33,9 @@ jobs:
         python -m pytest src/tests/
         
     - name: Upload model artifact
-      uses: actions/upload-artifact@v2
+
+      uses: actions/upload-artifact@v4
       with:
         name: trained-model
         path: mnist_project/src/model_mnist_*.pth 
+

README.md

@@ -0,0 +1,14 @@
+# ERA-V3 MNIST Project
+
+This project implements a CNN model for MNIST digit classification with CI/CD pipeline.
+
+## Project Structure
+- `mnist_project/`: Contains the main project code
+  - `src/`: Source code directory
+    - `model.py`: CNN model architecture
+    - `train.py`: Training script
+    - `utils.py`: Utility functions
+    - `tests/`: Test files
+
+## Local Setup
+1. Clone the repository: 

mnist_project/src/tests/test_training.py

@@ -1,9 +1,16 @@
 import torch
+import glob
 from src.model import MNISTModel
 from src.utils import evaluate_model
 
 def test_model_accuracy():
     model = MNISTModel()
-    model.load_state_dict(torch.load('model_mnist_latest.pth'))
+    # Find the most recent model file
+    model_files = glob.glob('model_mnist_*.pth')
+    if not model_files:
+        raise FileNotFoundError("No model file found")
+    latest_model = max(model_files)  # Gets the most recent file
+
+    model.load_state_dict(torch.load(latest_model))
     accuracy = evaluate_model(model)
     assert accuracy >= 0.8, f"Model accuracy {accuracy} is below 0.8"