Unverified Commit 4fce84a1 authored by Franz Pletz's avatar Franz Pletz Committed by GitHub
Browse files

Merge pull request #295759 from iynaix/bump-webdataset

parents 6e8d6c46 82d2bc1b
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -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"
+6 −4
Original line number Diff line number Diff line
@@ -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 ];
  };