Merge pull request #2 from BlazBert/main

Updated with the CI pipeline for testing model deployments.

Author: BlazBert

.github/workflows/ci.yml

@@ -0,0 +1,48 @@
+name: Test FastAPI NVG Endpoint
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test-nvg-api:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.10'
+
+    - name: Install test tools
+      run: pip install requests pytest numpy
+
+    - name: Build Docker image
+      run: docker build -t sl-nvg ./nvg-endpoint
+
+    - name: Run container
+      run: docker run -d -p 8000:8000 --name sl-nvg sl-nvg
+
+    - name: Wait for API to be Ready
+      run: |
+        for i in {1..10}; do
+          if curl -s http://localhost:8000/docs; then
+            echo "API is up!"
+            break
+          fi
+          echo "Waiting for API..."
+          sleep 3
+        done
+
+    - name: Run API test
+      run: pytest tests/
+
+    - name: Cleanup
+      run: |
+        docker stop sl-nvg
+        docker rm sl-nvg

README.md

@@ -113,3 +113,7 @@ This setup enables scalable deployment of your model API endpoint with built-in
 
 ## License
 GNU General Public License v3.0 see [LICENSE](LICENSE) file for details.
+
+## Acknowledgment
+
+This work was supported by funding from the European Union's Horizon Europe Framework Programme [NANCY](https://nancy-project.eu/) project under Grant Agreement No. 101096456.

loadtesting.py

@@ -4,6 +4,7 @@
 import pickle
 import numpy as np
 
+#added a commet to test something
 
 # Functions
 def make_request(url: str, data: List) -> Tuple[float, any]:

tests/test_api.py

@@ -0,0 +1,23 @@
+import requests
+from typing import List, Tuple
+import requests
+import pickle
+import numpy as np
+
+# Data
+data_url = 'nvg_inference_data.pkl'
+# labels = ["Normal", "SlowD", "SuddenD", "SuddenR", "InstaD"] 
+
+with open(data_url, 'rb') as f:
+    data_all = pickle.load(f)
+
+data_t = np.expand_dims(data_all[0], axis=0)
+
+import time
+
+def test_nvg_endpoint():
+    # Wait for the container to be ready
+    time.sleep(3)
+
+    response = requests.post("http://localhost:8000/NVG", json=data_t.tolist())
+    assert response.status_code == 200