From 75d1ac4b0b2fa429c30ba6a19c6952f3492fa9b1 Mon Sep 17 00:00:00 2001
From: Vignesh Raman <vignesh.raman@collabora.com>
Date: Mon, 7 Jul 2025 07:29:17 +0530
Subject: [PATCH] ci/lava: add main() function to fix entry point

Signed-off-by: Vignesh Raman <vignesh.raman@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/35961>
---
 .gitlab-ci/lava/lava_job_submitter.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/.gitlab-ci/lava/lava_job_submitter.py b/.gitlab-ci/lava/lava_job_submitter.py
index ee36f47b401..2a262313bb2 100755
--- a/.gitlab-ci/lava/lava_job_submitter.py
+++ b/.gitlab-ci/lava/lava_job_submitter.py
@@ -609,7 +609,7 @@ class StructuredLoggerWrapper:
         return context
 
 
-if __name__ == "__main__":
+def main():
     # given that we proxy from DUT -> LAVA dispatcher -> LAVA primary -> us ->
     # GitLab runner -> GitLab primary -> user, safe to say we don't need any
     # more buffering
@@ -617,3 +617,6 @@ if __name__ == "__main__":
     sys.stderr.reconfigure(line_buffering=True)
 
     fire.Fire(LAVAJobSubmitter)
+
+if __name__ == "__main__":
+    main()