feat(ci): add GitHub Action for Python tests, fix EarlyStopping logic, and add unit tests

.github/workflows/release.yml

@@ -36,7 +36,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install python-semantic-release==9.11.0  # Locking to a specific version
+        pip install python-semantic-release
 
     - name: Configure Git
       run: |

.github/workflows/tests.yml

@@ -0,0 +1,41 @@
+name: Python Application Tests
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    strategy:
+      matrix:
+        python-version: ['3.9', '3.10', '3.11', '3.12']
+
+    steps:
+      # Step 1: Checkout the repository
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      # Step 2: Set up Python
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: 'pip'
+
+      # Step 3: Install dependencies
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+
+      # Step 4: Install pytest (if not in requirements.txt)
+      - name: Install pytest
+        run: pip install pytest
+
+      # Step 5: Run tests
+      - name: Run tests
+        run: pytest

pytorchtools.py

@@ -3,14 +3,16 @@
 
 import numpy as np
 import torch
+
+
 class EarlyStopping:
     """Early stops the training if validation loss doesn't improve after a given patience."""
     def __init__(self, patience=7, verbose=False, delta=0, path='checkpoint.pt', trace_func=print):
         """
         Args:
             patience (int): How long to wait after last time validation loss improved.
                             Default: 7
-            verbose (bool): If True, prints a message for each validation loss improvement. 
+            verbose (bool): If True, prints a message for each validation loss improvement.
                             Default: False
             delta (float): Minimum change in the monitored quantity to qualify as an improvement.
                             Default: 0
@@ -22,31 +24,32 @@ def __init__(self, patience=7, verbose=False, delta=0, path='checkpoint.pt', tra
         self.patience = patience
         self.verbose = verbose
         self.counter = 0
-        self.best_score = None
+        self.best_val_loss = None
         self.early_stop = False
-        self.val_loss_min = np.Inf
+        self.val_loss_min = np.inf
         self.delta = delta
         self.path = path
         self.trace_func = trace_func
-    def __call__(self, val_loss, model):
 
-        score = -val_loss
+    def __call__(self, val_loss, model):
 
-        if self.best_score is None:
-            self.best_score = score
+        if self.best_val_loss is None:
+            self.best_val_loss = val_loss
             self.save_checkpoint(val_loss, model)
-        elif score < self.best_score + self.delta:
+        elif val_loss < self.best_val_loss - self.delta:
+            # Significant improvement detected
+            self.best_val_loss = val_loss
+            self.save_checkpoint(val_loss, model)
+            self.counter = 0  # Reset counter since improvement occurred
+        else:
+            # No significant improvement
             self.counter += 1
             self.trace_func(f'EarlyStopping counter: {self.counter} out of {self.patience}')
             if self.counter >= self.patience:
                 self.early_stop = True
-        else:
-            self.best_score = score
-            self.save_checkpoint(val_loss, model)
-            self.counter = 0
 
     def save_checkpoint(self, val_loss, model):
-        '''Saves model when validation loss decrease.'''
+        '''Saves model when validation loss decreases.'''
         if self.verbose:
             self.trace_func(f'Validation loss decreased ({self.val_loss_min:.6f} --> {val_loss:.6f}).  Saving model ...')
         torch.save(model.state_dict(), self.path)

requirements-dev.txt

@@ -0,0 +1,4 @@
+# requirements-dev.txt
+-r requirements.txt
+pytest
+pytest-mock

requirements.txt

@@ -1,3 +1,4 @@
+# requirements.txt
 matplotlib
 numpy
 torchvision