Merge pull request #233057 from mweinelt/onnx-1.14.0

{ lib

, stdenv

, buildPythonPackage

, python3

, bash

, cmake

, fetchFromGitHub

, gtest

, isPy27

, nbval

, numpy

, parameterized

, protobuf

, pybind11

, pytestCheckHook

, six

, pythonOlder

, tabulate

, typing-extensions

, pythonRelaxDepsHook

}:

let

  gtestStatic = gtest.override { static = true; };

in buildPythonPackage rec {

  pname = "onnx";

  version = "1.13.1";

  version = "1.14.0";

  format = "setuptools";

  disabled = isPy27;

  disabled = pythonOlder "3.8";

  src = fetchFromGitHub {

    owner = pname;

    repo = pname;

    rev = "refs/tags/v${version}";

    hash = "sha256-10MH23XpAv/uDW/2tRFGS2lKU8hnaNBwbIBIgVc7Jpk=";

    hash = "sha256-f+s25Y/jGosaSdoZY6PE3j6pENkfDcD+IQndrbtuzWg=";

  };

  nativeBuildInputs = [

    cmake

    pythonRelaxDepsHook

    pybind11

  ];

  pythonRelaxDeps = [ "protobuf" ];

  propagatedBuildInputs = [

    protobuf

    numpy

    six

    typing-extensions

  ];

  nativeCheckInputs = [

    nbval

    parameterized

    pytestCheckHook

    tabulate

  ];

    # Set CMAKE_INSTALL_LIBDIR to lib explicitly, because otherwise it gets set

    # to lib64 and cmake incorrectly looks for the protobuf library in lib64

    export CMAKE_ARGS="-DCMAKE_INSTALL_LIBDIR=lib -DONNX_USE_PROTOBUF_SHARED_LIBS=ON"

  '' + lib.optionalString doCheck ''

    export CMAKE_ARGS+=" -Dgoogletest_STATIC_LIBRARIES=${gtestStatic}/lib/libgtest.a -Dgoogletest_INCLUDE_DIRS=${lib.getDev gtestStatic}/include"

    export ONNX_BUILD_TESTS=1

  '';

  # The setup.py does all the configuration

  dontUseCmakeConfigure = true;

  doCheck = true;

  preCheck = ''

    export HOME=$(mktemp -d)

    # detecting source dir as a python package confuses pytest

    mv onnx/__init__.py onnx/__init__.py.hidden

  '';

  pytestFlagsArray = [ "onnx/test" "onnx/examples" ];

  pytestFlagsArray = [

    "onnx/test"

    "onnx/examples"

  ];

  disabledTests = [

    # attempts to fetch data from web

    "test_bvlc_alexnet_cpu"

    "test_squeezenet_cpu"

    "test_vgg19_cpu"

    "test_zfnet512_cpu"

  ] ++ lib.optionals stdenv.isAarch64 [

    # AssertionError: Output 0 of test 0 in folder

    "test__pytorch_converted_Conv2d_depthwise_padded"

    "test__pytorch_converted_Conv2d_dilated"

    "test_dft"

    "test_dft_axis"

    # AssertionError: Mismatch in test 'test_Conv2d_depthwise_padded'

    "test_xor_bcast4v4d"

    # AssertionError: assert 1 == 0

    "test_ops_tested"

  ];

  disabledTestPaths = [

    # Unexpected output fields from running code: {'stderr'}

    "onnx/examples/np_array_tensorproto.ipynb"

  ];

  __darwinAllowLocalNetworking = true;

  postCheck = ''

    # run "cpp" tests

    .setuptools-cmake-build/onnx_gtests