rename BaseCollector to DataCollector

KunBB · Apr 16, 2019 · 7de55cb · 7de55cb
1 parent 90cbd9c
commit 7de55cb
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/rlkit/core/rl_algorithm.py b/rlkit/core/rl_algorithm.py
@@ -5,7 +5,7 @@
 
 from rlkit.core import logger, eval_util
 from rlkit.data_management.replay_buffer import ReplayBuffer
-from rlkit.samplers.data_collector import BaseCollector
+from rlkit.samplers.data_collector import DataCollector
 
 
 def _get_epoch_timings():
@@ -27,8 +27,8 @@ def __init__(
             trainer,
             exploration_env,
             evaluation_env,
-            exploration_data_collector: BaseCollector,
-            evaluation_data_collector: BaseCollector,
+            exploration_data_collector: DataCollector,
+            evaluation_data_collector: DataCollector,
             replay_buffer: ReplayBuffer,
     ):
         self.trainer = trainer

diff --git a/rlkit/samplers/data_collector/__init__.py b/rlkit/samplers/data_collector/__init__.py
@@ -1,5 +1,5 @@
 from rlkit.samplers.data_collector.base import (
-    BaseCollector,
+    DataCollector,
     PathCollector,
     StepCollector,
 )

diff --git a/rlkit/samplers/data_collector/base.py b/rlkit/samplers/data_collector/base.py
@@ -1,7 +1,7 @@
 import abc
 
 
-class BaseCollector(object, metaclass=abc.ABCMeta):
+class DataCollector(object, metaclass=abc.ABCMeta):
     def end_epoch(self, epoch):
         pass
 
@@ -16,7 +16,7 @@ def get_epoch_paths(self):
         pass
 
 
-class PathCollector(BaseCollector, metaclass=abc.ABCMeta):
+class PathCollector(DataCollector, metaclass=abc.ABCMeta):
     @abc.abstractmethod
     def collect_new_paths(
             self,
@@ -27,7 +27,7 @@ def collect_new_paths(
         pass
 
 
-class StepCollector(BaseCollector, metaclass=abc.ABCMeta):
+class StepCollector(DataCollector, metaclass=abc.ABCMeta):
     @abc.abstractmethod
     def collect_new_steps(
             self,