refactor(main): replace print with logging
This commit is contained in:
parent
b70cd1c27b
commit
4151d2307e
1 changed files with 6 additions and 1 deletions
|
|
@ -1,4 +1,9 @@
|
||||||
|
import logging
|
||||||
|
|
||||||
import jax
|
import jax
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
print(jax.devices())
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
logger.info(f"JAX devices: {jax.devices()}")
|
||||||
|
|
|
||||||
Reference in a new issue