2019-10-25 19:28:44 +00:00
|
|
|
import attr
|
|
|
|
|
import pytest
|
|
|
|
|
import requests
|
|
|
|
|
from furl import furl
|
|
|
|
|
|
|
|
|
|
import six
|
2020-12-22 21:00:57 +00:00
|
|
|
from clearml_agent.helper.package.pytorch import PytorchRequirement
|
2019-10-25 19:28:44 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@attr.s
|
|
|
|
|
class PytorchURLWheel(object):
|
|
|
|
|
os = attr.ib()
|
|
|
|
|
cuda = attr.ib()
|
|
|
|
|
python = attr.ib()
|
|
|
|
|
pytorch = attr.ib()
|
|
|
|
|
url = attr.ib()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
wheels = [
|
|
|
|
|
PytorchURLWheel(os=os, cuda=cuda, python=python, pytorch=pytorch_version, url=url)
|
|
|
|
|
for os, os_d in PytorchRequirement.MAP.items()
|
|
|
|
|
for cuda, cuda_d in os_d.items()
|
|
|
|
|
if isinstance(cuda_d, dict)
|
|
|
|
|
for python, python_d in cuda_d.items()
|
|
|
|
|
if isinstance(python_d, dict)
|
|
|
|
|
for pytorch_version, url in python_d.items()
|
|
|
|
|
if isinstance(url, six.string_types) and furl(url).scheme
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.parametrize('wheel', wheels, ids=[','.join(map(str, attr.astuple(wheel))) for wheel in wheels])
|
|
|
|
|
def test_map(wheel):
|
|
|
|
|
assert requests.head(wheel.url).ok
|
