change k8s startup probe script to function call

2023-06-20 17:04:03 +02:00 · 2023-06-20 17:04:03 +02:00 · d63435e092
commit d63435e092
parent 7e995bd78b
1 changed files with 1 additions and 1 deletions
--- 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: