From bc0cc5fbd329cbdad61048fb82ba2bc0d9e1d914 Mon Sep 17 00:00:00 2001
From: alarv <alarvfm@gmail.com>
Date: Sun, 28 Jul 2024 13:15:49 +0300
Subject: [PATCH] fix: add healthcheck to 1

---
 .github/workflows/publish.yml | 10 +++++-----
 main.py                       |  3 +++
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 2fcfa5e..d36e582 100644
--- 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
index 77f795b..98dcf82 100644
--- 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):