refactor(main): replace print with logging

2026-03-31 19:51:17 +00:00 · 2026-03-31 19:51:17 +00:00 · 4151d2307e
commit 4151d2307e
parent b70cd1c27b
1 changed files with 6 additions and 1 deletions
--- 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()}")