swendel/nixpkgs
1
0
Fork 0
mirror of https://github.com/SebastianWendel/nixpkgs.git synced 2024-09-20 12:29:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #291751 from GaetanLepage/tensordict

Browse source 
python311Packages.tensordict: 0.3.0 -> 0.3.1
This commit is contained in:
Samuel Ainsworth 2024-02-29 11:45:46 -05:00 committed by GitHub
parent dd74935580 60fe64acc3
commit fdb669ba48
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
pkgs/development/python-modules/tensordict/default.nix
View file

@ -10,11 +10,12 @@
, numpy
, h5py
, pytestCheckHook
, stdenv
}:
buildPythonPackage rec {
pname = "tensordict";
version = "0.3.0";
version = "0.3.1";
pyproject = true;
disabled = pythonOlder "3.8";
@ -23,7 +24,7 @@ buildPythonPackage rec {
owner = "pytorch";
repo = "tensordict";
rev = "refs/tags/v${version}";
hash = "sha256-XTFUzPs/fqX3DPtu/qSE1hY+7r/HToPVPaTyVRzDT/E=";
hash = "sha256-eCx1r7goqOdGX/0mSGCiLhdGQTh4Swa5aFiLSsL56p0=";
};
nativeBuildInputs = [
@ -53,6 +54,18 @@ buildPythonPackage rec {
pytestCheckHook
];
# RuntimeError: internal error
disabledTests = lib.optionals (stdenv.hostPlatform.system == "aarch64-linux") [
"test_add_scale_sequence"
"test_modules"
"test_setattr"
];
# ModuleNotFoundError: No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package
disabledTestPaths = lib.optionals stdenv.isDarwin [
"test/test_distributed.py"
];
meta = with lib; {
description = "A pytorch dedicated tensor container";
changelog = "https://github.com/pytorch/tensordict/releases/tag/v${version}";