From 72e19ede28f901897d0fbeb022bdad6a18e6c6fe Mon Sep 17 00:00:00 2001 From: Alex Wu Date: Fri, 11 Sep 2020 12:43:43 -0700 Subject: [PATCH] [hotfix] accelerator_types (#10725) * . * . --- python/ray/util/accelerators/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/ray/util/accelerators/__init__.py b/python/ray/util/accelerators/__init__.py index b4dcc406a..b170a397f 100644 --- a/python/ray/util/accelerators/__init__.py +++ b/python/ray/util/accelerators/__init__.py @@ -1,4 +1,5 @@ -from ray.util.accelerators.accelerators import NVIDIA_TESLA_V100 +from ray.util.accelerators.accelerators import NVIDIA_TESLA_V100, \ + NVIDIA_TESLA_P100, NVIDIA_TESLA_T4, NVIDIA_TESLA_P4, NVIDIA_TESLA_K80 __all__ = [ "NVIDIA_TESLA_V100", "NVIDIA_TESLA_P100", "NVIDIA_TESLA_T4",