2021-09-03 05:05:04 +08:00
|
|
|
from tianshou import data, env, exploration, policy, trainer, utils
|
2020-09-02 13:03:32 +08:00
|
|
|
|
2023-08-25 23:40:56 +02:00
|
|
|
__version__ = "1.0.0"
|
2020-09-02 13:03:32 +08:00
|
|
|
|
2020-03-13 17:49:22 +08:00
|
|
|
__all__ = [
|
2020-09-12 15:39:01 +08:00
|
|
|
"env",
|
|
|
|
|
"data",
|
|
|
|
|
"utils",
|
|
|
|
|
"policy",
|
|
|
|
|
"trainer",
|
|
|
|
|
"exploration",
|
2020-03-13 17:49:22 +08:00
|
|
|
]
|
2023-10-27 20:15:58 +02:00
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|
