python310Packages.tf2onnx: init at 1.14.0

authored by happysalada and committed by Yt da5286cd 78386f6b

+91 -2
+86
pkgs/development/python-modules/tf2onnx/default.nix
··· 1 + { lib 2 + , buildPythonPackage 3 + , fetchFromGitHub 4 + , pythonRelaxDepsHook 5 + , pytest-runner 6 + # runtime dependencies 7 + , numpy 8 + , onnx 9 + , requests 10 + , six 11 + , flatbuffers 12 + , protobuf 13 + , tensorflow 14 + # check dependencies 15 + , pytestCheckHook 16 + , graphviz 17 + , parameterized 18 + , pytest-cov 19 + , pyyaml 20 + , timeout-decorator 21 + , onnxruntime 22 + , keras 23 + }: 24 + 25 + buildPythonPackage rec { 26 + pname = "tf2onnx"; 27 + version = "1.14.0"; 28 + format = "setuptools"; 29 + 30 + src = fetchFromGitHub { 31 + owner = "onnx"; 32 + repo = "tensorflow-onnx"; 33 + rev = "v${version}"; 34 + hash = "sha256-JpXwf+GLjn0krsb5KnEhVuemWa0V2+wF10neDsdtOfI="; 35 + }; 36 + 37 + nativeBuildInputs = [ 38 + pythonRelaxDepsHook 39 + pytest-runner 40 + ]; 41 + 42 + pythonRelaxDeps = [ 43 + "flatbuffers" 44 + ]; 45 + 46 + propagatedBuildInputs = [ 47 + numpy 48 + onnx 49 + requests 50 + six 51 + flatbuffers 52 + protobuf 53 + tensorflow 54 + onnxruntime 55 + ]; 56 + 57 + pythonImportsCheck = [ "tf2onnx" ]; 58 + 59 + nativeCheckInputs = [ 60 + pytestCheckHook 61 + graphviz 62 + parameterized 63 + pytest-cov 64 + pyyaml 65 + timeout-decorator 66 + keras 67 + ]; 68 + 69 + # TODO investigate the failures 70 + disabledTestPaths = [ 71 + "tests/test_backend.py" 72 + "tests/test_einsum_helper.py" 73 + "tests/test_einsum_optimizers.py" 74 + ]; 75 + 76 + disabledTests = [ 77 + "test_profile_conversion_time" 78 + ]; 79 + 80 + meta = with lib; { 81 + description = "Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX"; 82 + homepage = "https://github.com/onnx/tensorflow-onnx"; 83 + license = licenses.asl20; 84 + maintainers = with maintainers; [ happysalada ]; 85 + }; 86 + }
+3 -2
pkgs/development/python-modules/transformers/default.nix
··· 18 18 , scikit-learn 19 19 , tensorflow 20 20 , onnxconverter-common 21 + , tf2onnx 21 22 , torch 22 23 , accelerate 23 24 , faiss ··· 99 100 tf = [ 100 101 tensorflow 101 102 onnxconverter-common 102 - # tf2onnx 103 + tf2onnx 103 104 # tensorflow-text 104 105 # keras-nlp 105 106 ]; ··· 119 120 ]; 120 121 onnx = [ 121 122 onnxconverter-common 122 - # tf2onnx 123 + tf2onnx 123 124 onnxruntime 124 125 ]; 125 126 modelcreation = [
+2
pkgs/top-level/python-packages.nix
··· 12387 12387 12388 12388 textx = callPackage ../development/python-modules/textx { }; 12389 12389 12390 + tf2onnx = callPackage ../development/python-modules/tf2onnx { }; 12391 + 12390 12392 tflearn = callPackage ../development/python-modules/tflearn { }; 12391 12393 12392 12394 tftpy = callPackage ../development/python-modules/tftpy { };