Commit 378940f7 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.tensordict: 0.10.0 -> 0.11.0 

Diff: https://github.com/pytorch/tensordict/compare/v0.10.0...v0.11.0

Changelog: https://github.com/pytorch/tensordict/releases/tag/v0.11.0
parent 45cb6642

pkgs/development/python-modules/tensordict/default.nix

+11 −6
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ 
  stdenv,

  buildPythonPackage,

  fetchFromGitHub,

  pythonAtLeast,



  # build-system

  pybind11,
@@ -27,16 +28,16 @@ 
  pytestCheckHook,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "tensordict";

  version = "0.10.0";

  version = "0.11.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "pytorch";

    repo = "tensordict";

    tag = "v${version}";

    hash = "sha256-yxyA9BfN2hp1C3s+g2zBM2gVtckH3LV7luWw8DshFUs=";

    tag = "v${finalAttrs.version}";

    hash = "sha256-PUPDKv10Ks4B1kpgbRcnmfWFUkpFEdxMmTNztFVfdK4=";

  };



  postPatch = ''
@@ -86,6 +87,10 @@ buildPythonPackage rec { 
    # hangs forever on some CPUs

    "test_map_iter_interrupt_early"

  ]

  ++ lib.optionals (pythonAtLeast "3.14") [

    # AssertionError: assert 'a string!' == 'a metadata!'

    "test_save_load_memmap"

  ]

  ++ lib.optionals stdenv.hostPlatform.isDarwin [

    # Hangs due to the use of a pool

    "test_chunksize_num_chunks"
@@ -109,9 +114,9 @@ buildPythonPackage rec { 


  meta = {

    description = "Pytorch dedicated tensor container";

    changelog = "https://github.com/pytorch/tensordict/releases/tag/${src.tag}";

    changelog = "https://github.com/pytorch/tensordict/releases/tag/${finalAttrs.src.tag}";

    homepage = "https://github.com/pytorch/tensordict";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}
 
})