ray/rllib/execution/__init__.py

from ray.rllib.execution.concurrency_ops import Concurrently, Enqueue, Dequeue
from ray.rllib.execution.learner_thread import LearnerThread
from ray.rllib.execution.metric_ops import (
    StandardMetricsReporting,
    CollectMetrics,
    OncePerTimeInterval,
    OncePerTimestepsElapsed,
)
from ray.rllib.execution.multi_gpu_learner_thread import MultiGPULearnerThread
from ray.rllib.execution.minibatch_buffer import MinibatchBuffer
from ray.rllib.execution.replay_ops import (
    StoreToReplayBuffer,
    Replay,
    SimpleReplayBuffer,
    MixInReplay,
)
from ray.rllib.execution.rollout_ops import (
    ParallelRollouts,
    AsyncGradients,
    ConcatBatches,
    SelectExperiences,
    StandardizeFields,
    synchronous_parallel_sample,
)
from ray.rllib.execution.train_ops import (
    TrainOneStep,
    MultiGPUTrainOneStep,
    ComputeGradients,
    ApplyGradients,
    AverageGradients,
    UpdateTargetNetwork,
    train_one_step,
)

__all__ = [
    "synchronous_parallel_sample",
    "train_one_step",
    "ApplyGradients",
    "AsyncGradients",
    "AverageGradients",
    "CollectMetrics",
    "ComputeGradients",
    "ConcatBatches",
    "Concurrently",
    "Dequeue",
    "Enqueue",
    "LearnerThread",
    "MixInReplay",
    "MultiGPULearnerThread",
    "OncePerTimeInterval",
    "OncePerTimestepsElapsed",
    "ParallelRollouts",
    "Replay",
    "SelectExperiences",
    "SimpleReplayBuffer",
    "StandardMetricsReporting",
    "StandardizeFields",
    "StoreToReplayBuffer",
    "TrainOneStep",
    "MultiGPUTrainOneStep",
    "UpdateTargetNetwork",
    "MinibatchBuffer",
]