# -*- coding: utf-8; mode: tcl; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- vim:fenc=utf-8:ft=tcl:et:sw=4:ts=4:sts=4

PortSystem          1.0
PortGroup           python 1.0
PortGroup           github 1.0

github.setup        pytorch vision 0.15.2 v
# Change github.tarball_from to 'releases' or 'archive' next update
github.tarball_from tarball
revision            0
name                py-torchvision

license             BSD
maintainers         nomaintainer
supported_archs     arm64 x86_64

description         PyTorch datasets, transforms and models specific to \
                    computer vision

long_description    The torchvision package consists of popular \
                    datasets, model architectures, and common image \
                    transformations for computer vision.

checksums           rmd160  a52c8e235cfd977d78db5f4666f518dc42565fb6 \
                    sha256  e7425cb4313919d9b3873c8395d1e7d7abb9e0f32a38ce9dcb3b4160ffd979b8 \
                    size    13091302

python.versions     39 310

if {${os.platform} eq "darwin" && ${os.major} <= 15} {
    known_fail yes
    pre-fetch {
        ui_error "$subport is not supported on OS X 10.11 and older."
        return -code error "unsupported platform"
    }
}

if {${name} ne ${subport}} {
    depends_lib-append \
                    port:py${python.version}-Pillow \
                    port:py${python.version}-pytorch \
                    port:zmq

    compiler.cxx_standard 2014

    # https://github.com/pytorch/vision/blob/master/packaging/build_wheel.sh
    build.cmd       ${python.bin} setup.py
    build.target    bdist_wheel

    test.run        yes
}
