From 4151d2307e07a8b50ce55a13af99711b78753806 Mon Sep 17 00:00:00 2001 From: Tibo De Peuter Date: Tue, 31 Mar 2026 19:51:17 +0000 Subject: [PATCH] refactor(main): replace print with logging --- src/main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index ef7c36e..ac40465 100644 --- a/src/main.py +++ b/src/main.py @@ -1,4 +1,9 @@ +import logging + import jax +logger = logging.getLogger(__name__) + if __name__ == "__main__": - print(jax.devices()) + logging.basicConfig(level=logging.INFO) + logger.info(f"JAX devices: {jax.devices()}")