Merge pull request #35 from springfall2008/update

Version and test flow

Author: springfall2008

.github/workflows/test.yml

@@ -0,0 +1,47 @@
+name: PredAI Tests
+
+on:
+  pull_request:
+    branches: [ main, master, develop ]
+  push:
+    branches: [ main, master, develop ]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v4
+      
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.12'
+        
+    - name: Cache pip packages
+      uses: actions/cache@v4
+      with:
+        path: ~/.cache/pip
+        key: ${{ runner.os }}-pip-${{ hashFiles('predai/requirements.txt') }}
+        restore-keys: |
+          ${{ runner.os }}-pip-
+          
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r predai/requirements.txt
+        
+    - name: Run unit tests
+      run: |
+        python -m unittest test_predai.py -v
+        
+    - name: Test summary
+      if: always()
+      run: |
+        if [ $? -eq 0 ]; then
+          echo "✅ All tests passed successfully!"
+        else
+          echo "❌ Some tests failed"
+          exit 1
+        fi

.gitignore

@@ -19,6 +19,10 @@ wheels/
 *.egg-info/
 .installed.cfg
 *.egg
+.lr_find*
+.pytest_cache/
+events.out.*
+lightning_logs/*
 
 # Virtual environments
 venv/

predai/requirements.txt

@@ -1,4 +1,4 @@
-neuralprophet==0.9.0
+neuralprophet
 requests
 aiohttp
 pyyaml

predai/rootfs/predai.py

@@ -13,9 +13,10 @@
 import math
 import yaml
 
-# Fix for PyTorch 2.6 weights_only=True default
+# Fix for PyTorch 2.6+ weights_only=True default
 # Add NeuralProphet classes to safe globals for checkpoint loading
 try:
+    import torch
     import torch.serialization
     from neuralprophet.configure import (
         ConfigSeasonality,
@@ -27,6 +28,7 @@
         ConfigEvents,
         ConfigCountryHolidays,
     )
+    from collections import OrderedDict
 
     # Add all NeuralProphet configuration classes to safe globals
     torch.serialization.add_safe_globals([
@@ -38,7 +40,17 @@
         ConfigLagged,
         ConfigEvents,
         ConfigCountryHolidays,
+        OrderedDict,
     ])
+    
+    # Monkey-patch torch.load to use weights_only=False by default
+    # This is needed for PyTorch 2.6+ compatibility with PyTorch Lightning
+    _original_torch_load = torch.load
+    def _patched_torch_load(*args, **kwargs):
+        if 'weights_only' not in kwargs:
+            kwargs['weights_only'] = False
+        return _original_torch_load(*args, **kwargs)
+    torch.load = _patched_torch_load
 except (ImportError, AttributeError):
     # If torch.serialization or classes are not available, continue without the fix
     # This allows backward compatibility with older PyTorch versions
@@ -483,4 +495,5 @@ async def main():
                 break
             await asyncio.sleep(60)
 
-asyncio.run(main())
+if __name__ == "__main__":
+    asyncio.run(main())

run_all

@@ -0,0 +1,33 @@
+#!/bin/bash
+# Run all PredAI unit tests
+
+set -e
+
+# Check if virtual environment exists
+if [ ! -d "venv" ]; then
+    echo "Error: Virtual environment not found!"
+    echo "Please run ./setup.csh first to create the environment"
+    exit 1
+fi
+
+# Activate virtual environment
+echo "Activating virtual environment..."
+source venv/bin/activate
+
+# Run the tests
+echo ""
+echo "Running PredAI unit tests..."
+echo "================================"
+python -m unittest test_predai.py -v
+
+# Check test result
+if [ $? -eq 0 ]; then
+    echo ""
+    echo "================================"
+    echo "✅ All tests passed successfully!"
+else
+    echo ""
+    echo "================================"
+    echo "❌ Some tests failed"
+    exit 1
+fi

setup.csh

@@ -0,0 +1,21 @@
+# Setup script for PredAI test environment
+# Creates a virtual environment and installs dependencies
+
+echo "Creating Python virtual environment..."
+python3 -m venv venv
+
+echo "Activating virtual environment..."
+source venv/bin/activate
+
+echo "Installing dependencies..."
+python -m pip install --upgrade pip
+python -m pip install -r predai/requirements.txt
+
+echo ""
+echo "Setup complete!"
+echo "To activate the environment manually, run:"
+echo "  source venv/bin/activate.csh"
+echo ""
+echo "To run tests, execute:"
+echo "  ./run_all"
+