mirror of
https://github.com/clearml/clearml-agent
synced 2025-01-31 09:06:52 +00:00
34 lines
872 B
Python
34 lines
872 B
Python
import attr
|
|
import pytest
|
|
import requests
|
|
from furl import furl
|
|
|
|
import six
|
|
from trains_agent.helper.package.pytorch import PytorchRequirement
|
|
|
|
|
|
@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
|