chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,22 @@
|
||||
import sys
|
||||
|
||||
import pytest
|
||||
|
||||
from ray.util import accelerators
|
||||
from ray.util.annotations import RayDeprecationWarning
|
||||
|
||||
|
||||
def test_accelerators():
|
||||
assert accelerators.NVIDIA_TESLA_K80 == "K80"
|
||||
assert accelerators.NVIDIA_A100 == "A100"
|
||||
with pytest.raises(
|
||||
AttributeError,
|
||||
match="module 'ray.util.accelerators' has no attribute 'NVIDIA_INVALID'",
|
||||
):
|
||||
_ = accelerators.NVIDIA_INVALID
|
||||
with pytest.warns(RayDeprecationWarning):
|
||||
assert accelerators.NVIDIA_TESLA_A100 == "A100"
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(pytest.main(["-sv", __file__]))
|
||||
Reference in New Issue
Block a user