25 lines
454 B
Python
25 lines
454 B
Python
from tianshou import data, env, exploration, policy, trainer, utils
|
|
|
|
__version__ = "1.0.0"
|
|
|
|
__all__ = [
|
|
"env",
|
|
"data",
|
|
"utils",
|
|
"policy",
|
|
"trainer",
|
|
"exploration",
|
|
]
|
|
|
|
|
|
def _configure_logging() -> None:
|
|
from sensai.util import logging
|
|
|
|
def logging_configure_callback() -> None:
|
|
logging.getLogger("numba").setLevel(logging.INFO)
|
|
|
|
logging.set_configure_callback(logging_configure_callback)
|
|
|
|
|
|
_configure_logging()
|