Source code for garage.sampler.base

import abc


class Sampler(abc.ABC):
    @abc.abstractmethod
    def start_worker(self):
        """Initialize the sampler,

        e.g. launching parallel workers if necessary.
        """
        pass

    @abc.abstractmethod
    def obtain_samples(self, itr):
        """Collect samples for the given iteration number.

        Args:
            itr (int): Iteration number.

        Returns:
            list[dict]: A list of paths.

        """
        pass

    @abc.abstractmethod
    def shutdown_worker(self):
        """Terminate workers if necessary."""
        pass


class BaseSampler(Sampler):
    """Base class for sampler.

    Args:
        algo (garage.np.algos.RLAlgorithm): The algorithm.
        env (gym.Env): The environment.

    """

    def __init__(self, algo, env):
        self.algo = algo
        self.env = env