wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-16 11:27:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib] Remove "Common", cleanup some code (#2348)

Browse Source
This commit is contained in:
Richard Liaw
2018-07-08 13:03:53 -07:00
committed by GitHub
parent 1d05cd7077
commit 4d7da9f668
15 changed files with 58 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/agents/bc/bc_evaluator.py
+2 -2
View File
@@ -8,11 +8,11 @@ from six.moves import queue
import ray
from ray.rllib.agents.bc.experience_dataset import ExperienceDataset
from ray.rllib.agents.bc.policy import BCPolicy
from ray.rllib.evaluation.interface import PolicyEvaluator
from ray.rllib.evaluation.interface import EvaluatorInterface
from ray.rllib.models import ModelCatalog
class BCEvaluator(PolicyEvaluator):
class BCEvaluator(EvaluatorInterface):
def __init__(self, env_creator, config, logdir):
env = ModelCatalog.get_preprocessor_as_wrapper(env_creator(
config["env_config"]), config["model"])