Commit 0b335770 authored by dmayle's avatar dmayle Committed by Weijia Wang
Browse files

python310Packages.tensorflow-datasets: fix build

parent 3df03f23

pkgs/development/python-modules/tensorflow-datasets/default.nix

+24 −2
Original line number Diff line number Diff line
@@ -2,6 +2,8 @@ 
, attrs

, beautifulsoup4

, buildPythonPackage

, click

, datasets

, dill

, dm-tree

, fetchFromGitHub
@@ -14,6 +16,7 @@ 
, jinja2

, langdetect

, lib

, lxml

, matplotlib

, mwparserfromhell

, networkx
@@ -24,6 +27,7 @@ 
, pillow

, promise

, protobuf

, psutil

, pycocotools

, pydub

, pytest-xdist
@@ -65,6 +69,7 @@ buildPythonPackage rec { 
    numpy

    promise

    protobuf

    psutil

    requests

    six

    tensorflow-metadata
@@ -79,12 +84,15 @@ buildPythonPackage rec { 
  nativeCheckInputs = [

    apache-beam

    beautifulsoup4

    click

    datasets

    ffmpeg

    imagemagick

    jax

    jaxlib

    jinja2

    langdetect

    lxml

    matplotlib

    mwparserfromhell

    networkx
@@ -109,19 +117,29 @@ buildPythonPackage rec { 
    "tensorflow_datasets/core/dataset_info_test.py"

    "tensorflow_datasets/core/features/features_test.py"

    "tensorflow_datasets/core/github_api/github_path_test.py"

    "tensorflow_datasets/core/registered_test.py"

    "tensorflow_datasets/core/utils/gcs_utils_test.py"

    "tensorflow_datasets/import_without_tf_test.py"

    "tensorflow_datasets/scripts/cli/build_test.py"



    # Requires `pretty_midi` which is not packaged in `nixpkgs`.

    "tensorflow_datasets/audio/groove_test.py"

    "tensorflow_datasets/audio/groove.py"

    "tensorflow_datasets/datasets/groove/groove_dataset_builder_test.py"



    # Requires `crepe` which is not packaged in `nixpkgs`.

    "tensorflow_datasets/audio/nsynth_test.py"

    "tensorflow_datasets/audio/nsynth.py"

    "tensorflow_datasets/datasets/nsynth/nsynth_dataset_builder_test.py"



    # Requires `conllu` which is not packaged in `nixpkgs`.

    "tensorflow_datasets/core/dataset_builders/conll/conllu_dataset_builder_test.py"

    "tensorflow_datasets/datasets/universal_dependencies/universal_dependencies_dataset_builder_test.py"

    "tensorflow_datasets/datasets/xtreme_pos/xtreme_pos_dataset_builder_test.py"



    # Requires `gcld3` and `pretty_midi` which are not packaged in `nixpkgs`.

    "tensorflow_datasets/core/lazy_imports_lib_test.py"



    # Requires `tensorflow_io` which is not packaged in `nixpkgs`.

    "tensorflow_datasets/core/features/audio_feature_test.py"

    "tensorflow_datasets/image/lsun_test.py"



    # Requires `envlogger` which is not packaged in `nixpkgs`.
@@ -133,6 +151,10 @@ buildPythonPackage rec { 
    # to the differences in some of the dependencies?

    "tensorflow_datasets/rl_unplugged/rlu_atari/rlu_atari_test.py"



    # Fails with `ValueError: setting an array element with a sequence`

    "tensorflow_datasets/core/dataset_utils_test.py"

    "tensorflow_datasets/core/features/sequence_feature_test.py"



    # Requires `tensorflow_docs` which is not packaged in `nixpkgs` and the test is for documentation anyway.

    "tensorflow_datasets/scripts/documentation/build_api_docs_test.py"