Merge pull request #295759 from iynaix/bump-webdataset

This commit is contained in:
Franz Pletz 2024-05-13 00:30:07 +02:00 committed by GitHub
commit 4fce84a169
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 4 deletions

View File

@ -69,6 +69,8 @@ buildPythonPackage rec {
"test_inference_with_data"
"test_pretrained_text_encoder"
"test_training_mt5"
# fails due to type errors
"test_num_shards"
] ++ lib.optionals (stdenv.isAarch64 && stdenv.isLinux) [
"test_training"
"test_training_coca"

View File

@ -16,14 +16,14 @@
}:
buildPythonPackage rec {
pname = "webdataset";
version = "0.2.86";
version = "0.2.88";
pyproject = true;
src = fetchFromGitHub {
owner = "webdataset";
repo = "webdataset";
rev = "refs/tags/${version}";
hash = "sha256-aTjxoSoQ9LH4gcFmV+7Aj0HNIpvsFHTrxFUpAtB3nkM=";
hash = "sha256-wsBOBUK4VIGMQXYdgbgsuSH4XYvxDsUv3rh0S5rvA6c=";
};
nativeBuildInputs = [
@ -51,16 +51,17 @@ buildPythonPackage rec {
disabledTests = [
# requires network
"test_batched"
"test_cache_dir"
"test_concurrent_download_and_open"
"test_dataloader"
"test_decode_handlers"
"test_decoder"
"test_decoder"
"test_download"
"test_handlers"
"test_pipe"
"test_remote_file"
"test_shard_syntax"
"test_torchvision"
"test_torchvision"
"test_unbatched"
"test_yaml3"
] ++ lib.optionals stdenv.isDarwin [
@ -78,6 +79,7 @@ buildPythonPackage rec {
description = "A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch";
mainProgram = "widsindex";
homepage = "https://github.com/webdataset/webdataset";
changelog = "https://github.com/webdataset/webdataset/releases/tag/${version}";
license = licenses.bsd3;
maintainers = with maintainers; [ iynaix ];
};