mirror of
https://github.com/gryf/coach.git
synced 2025-12-17 11:10:20 +01:00
Remove assertion from BatchRLGraphManager
This commit is contained in:
@@ -94,8 +94,6 @@ class BatchRLGraphManager(BasicRLGraphManager):
|
|||||||
self.schedule_params = schedule_params
|
self.schedule_params = schedule_params
|
||||||
|
|
||||||
def _create_graph(self, task_parameters: TaskParameters) -> Tuple[List[LevelManager], List[Environment]]:
|
def _create_graph(self, task_parameters: TaskParameters) -> Tuple[List[LevelManager], List[Environment]]:
|
||||||
assert self.agent_params.memory.load_memory_from_file_path or self.env_params, \
|
|
||||||
"BatchRL requires either a dataset to train from or an environment to collect a dataset from. "
|
|
||||||
if self.env_params:
|
if self.env_params:
|
||||||
# environment loading
|
# environment loading
|
||||||
self.env_params.seed = task_parameters.seed
|
self.env_params.seed = task_parameters.seed
|
||||||
|
|||||||
Reference in New Issue
Block a user