ray/rllib/models/tf/visionnet_v2.py

8 lines
238 B
Python
Raw Normal View History

from ray.rllib.models.tf.vision_net import VisionNetwork as TFVision
from ray.rllib.utils.deprecation import renamed_class
VisionNetwork = renamed_class(
cls=TFVision,
old_name="ray.rllib.models.tf.visionnet_v2.VisionNetwork",
)