fix: add healthcheck to 1

ntua-unit-of-control-and-informatics · Jul 28, 2024 · bc0cc5f · bc0cc5f
1 parent 29e7f51
commit bc0cc5f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -3,11 +3,11 @@ name: Publish Docker image
 on:
   push:
     branches: [
-      "release/sk-learn0.20",
-      "release/sk-learn0.22",
-      "release/sk-learn0.23",
-      "release/sk-learn0.24",
-      "release/sk-learn1",
+      "release/sk-learn0.20*",
+      "release/sk-learn0.22*",
+      "release/sk-learn0.23*",
+      "release/sk-learn0.24*",
+      "release/sk-learn1*",
     ]
 
 jobs:

diff --git a/main.py b/main.py
@@ -18,6 +18,9 @@
     allow_headers=["*"]
 )
 
+@app.get('/')
+def health_check():
+    return {'status': 'UP'}
 
 @app.post("/predict/sklearn/")
 # def sklearn_model(req: PredictionRequest):