refactor dir structure

This commit is contained in:
Dominik Moritz Roth 2024-06-02 11:13:04 +02:00
parent 8a013a4746
commit b34224f189
4 changed files with 3 additions and 2 deletions

View File

@ -4,6 +4,6 @@ try:
except ImportError:
pass
from fancy_rl.ppo import PPO
from fancy_rl.algos import PPO
__all__ = ["PPO"]

View File

@ -0,0 +1 @@
from fancy_rl.algos.ppo import PPO

View File

@ -2,7 +2,7 @@ import torch
from torchrl.modules import ActorValueOperator, ProbabilisticActor
from torchrl.objectives import ClipPPOLoss
from torchrl.objectives.value.advantages import GAE
from fancy_rl.on_policy import OnPolicy
from fancy_rl.algos.on_policy import OnPolicy
from fancy_rl.policy import Actor, Critic, SharedModule
class PPO(OnPolicy):