ray/rllib/policy/__init__.py

17 lines
502 B
Python

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from ray.rllib.policy.policy import Policy
from ray.rllib.policy.torch_policy import TorchPolicy
from ray.rllib.policy.tf_policy import TFPolicy
from ray.rllib.policy.torch_policy_template import build_torch_policy
from ray.rllib.policy.tf_policy_template import build_tf_policy
__all__ = [
"Policy",
"TFPolicy",
"TorchPolicy",
"build_tf_policy",
"build_torch_policy",
]