Updates across the board
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from retro_gamer.metadata import GameMetadata
|
||||
from retro_gamer.env import GameEnvironment
|
||||
from retro_gamer.trainer import DQNTrainer
|
||||
from retro_gamer.model_agent import TrainedPolicy, PolicyInput
|
||||
|
||||
__all__ = ["GameMetadata", "GameEnvironment", "DQNTrainer"]
|
||||
__all__ = ["GameMetadata", "GameEnvironment", "DQNTrainer", "TrainedPolicy", "PolicyInput"]
|
||||
|
||||
Reference in New Issue
Block a user