From d63435e092657f4ff152a1e7cdd02627ff45ad67 Mon Sep 17 00:00:00 2001
From: "francisco.schulz" <francisco.schulz@iqser.com>
Date: Tue, 20 Jun 2023 17:04:03 +0200
Subject: [PATCH] change k8s startup probe script to function call

---
 pyinfra/k8s_probes/startup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyinfra/k8s_probes/startup.py b/pyinfra/k8s_probes/startup.py
index 72b8c54..986bbe8 100644
--- a/pyinfra/k8s_probes/startup.py
+++ b/pyinfra/k8s_probes/startup.py
@@ -29,7 +29,7 @@ def check_token_file():
         return False
 
 
-if __name__ == "__main__":
+def run_checks():
     if check_token_file():
         sys.exit(0)
     else: