Depend on sensAI instead of copying its utils (logging, string)

.gitignore

@@ -153,6 +153,5 @@ videos/
 # might be needed for IDE plugins that can't read ruff config
 .flake8
 
-# temporary scripts (for ad-hoc testing), temp folder
+# folder for temporary files
 /temp
-/temp*.py

examples/atari/atari_dqn_hl.py

@@ -2,6 +2,9 @@
 
 import os
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.atari.atari_callbacks import (
     TestEpochCallbackDQNSetEps,
     TrainEpochCallbackNatureDQNEpsLinearDecay,
@@ -20,8 +23,6 @@
 from tianshou.highlevel.params.policy_wrapper import (
     PolicyWrapperFactoryIntrinsicCuriosity,
 )
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/atari/atari_iqn_hl.py

@@ -3,6 +3,9 @@
 import os
 from collections.abc import Sequence
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.atari.atari_callbacks import (
     TestEpochCallbackDQNSetEps,
     TrainEpochCallbackNatureDQNEpsLinearDecay,
@@ -17,8 +20,6 @@
     IQNExperimentBuilder,
 )
 from tianshou.highlevel.params.policy_params import IQNParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/atari/atari_ppo_hl.py

@@ -3,6 +3,9 @@
 import os
 from collections.abc import Sequence
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.atari.atari_network import (
     ActorFactoryAtariDQN,
     IntermediateModuleFactoryAtariDQNFeatures,
@@ -18,8 +21,6 @@
 from tianshou.highlevel.params.policy_wrapper import (
     PolicyWrapperFactoryIntrinsicCuriosity,
 )
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/atari/atari_sac_hl.py

@@ -2,6 +2,9 @@
 
 import os
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.atari.atari_network import (
     ActorFactoryAtariDQN,
     IntermediateModuleFactoryAtariDQNFeatures,
@@ -17,8 +20,6 @@
 from tianshou.highlevel.params.policy_wrapper import (
     PolicyWrapperFactoryIntrinsicCuriosity,
 )
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_a2c_hl.py

@@ -4,6 +4,8 @@
 from collections.abc import Sequence
 from typing import Literal
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
 from torch import nn
 
 from examples.mujoco.mujoco_env import MujocoEnvFactory
@@ -15,8 +17,6 @@
 from tianshou.highlevel.optim import OptimizerFactoryRMSprop
 from tianshou.highlevel.params.lr_scheduler import LRSchedulerFactoryLinear
 from tianshou.highlevel.params.policy_params import A2CParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_ddpg_hl.py

@@ -3,6 +3,9 @@
 import os
 from collections.abc import Sequence
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
 from tianshou.highlevel.experiment import (
@@ -11,8 +14,6 @@
 )
 from tianshou.highlevel.params.noise import MaxActionScaledGaussian
 from tianshou.highlevel.params.policy_params import DDPGParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_npg_hl.py

@@ -5,6 +5,8 @@
 from typing import Literal
 
 import torch
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
 
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
@@ -17,8 +19,6 @@
 )
 from tianshou.highlevel.params.lr_scheduler import LRSchedulerFactoryLinear
 from tianshou.highlevel.params.policy_params import NPGParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_ppo_hl.py

@@ -5,6 +5,8 @@
 from typing import Literal
 
 import torch
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
 
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
@@ -17,8 +19,6 @@
 )
 from tianshou.highlevel.params.lr_scheduler import LRSchedulerFactoryLinear
 from tianshou.highlevel.params.policy_params import PPOParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_redq_hl.py

@@ -4,6 +4,9 @@
 from collections.abc import Sequence
 from typing import Literal
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
 from tianshou.highlevel.experiment import (
@@ -12,8 +15,6 @@
 )
 from tianshou.highlevel.params.alpha import AutoAlphaFactoryDefault
 from tianshou.highlevel.params.policy_params import REDQParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_reinforce_hl.py

@@ -5,6 +5,8 @@
 from typing import Literal
 
 import torch
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
 
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
@@ -14,8 +16,6 @@
 )
 from tianshou.highlevel.params.lr_scheduler import LRSchedulerFactoryLinear
 from tianshou.highlevel.params.policy_params import PGParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_sac_hl.py

@@ -3,6 +3,9 @@
 import os
 from collections.abc import Sequence
 
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
+
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
 from tianshou.highlevel.experiment import (
@@ -11,8 +14,6 @@
 )
 from tianshou.highlevel.params.alpha import AutoAlphaFactoryDefault
 from tianshou.highlevel.params.policy_params import SACParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_td3_hl.py

@@ -4,6 +4,8 @@
 from collections.abc import Sequence
 
 import torch
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
 
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
@@ -16,8 +18,6 @@
     MaxActionScaledGaussian,
 )
 from tianshou.highlevel.params.policy_params import TD3Params
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(

examples/mujoco/mujoco_trpo_hl.py

@@ -5,6 +5,8 @@
 from typing import Literal
 
 import torch
+from sensai.util import logging
+from sensai.util.logging import datetime_tag
 
 from examples.mujoco.mujoco_env import MujocoEnvFactory
 from tianshou.highlevel.config import SamplingConfig
@@ -17,8 +19,6 @@
 )
 from tianshou.highlevel.params.lr_scheduler import LRSchedulerFactoryLinear
 from tianshou.highlevel.params.policy_params import TRPOParams
-from tianshou.utils import logging
-from tianshou.utils.logging import datetime_tag
 
 
 def main(