diff --git a/pkgs/development/python-modules/hoptorch/default.nix b/pkgs/development/python-modules/hoptorch/default.nix new file mode 100644 index 000000000000..791367818bd8 --- /dev/null +++ b/pkgs/development/python-modules/hoptorch/default.nix @@ -0,0 +1,59 @@ +{ + lib, + stdenv, + buildPythonPackage, + fetchFromGitHub, + + # tests + setuptools, + + # dependencies + pyvers, + torch, + + # tests + llvmPackages, + pytestCheckHook, +}: + +buildPythonPackage (finalAttrs: { + pname = "hoptorch"; + version = "0.1.4"; + pyproject = true; + __structuredAttrs = true; + + src = fetchFromGitHub { + owner = "vmoens"; + repo = "hoptorch"; + tag = "v${finalAttrs.version}"; + hash = "sha256-rhX81MidgltQ2YQtUdYoK1Qtz7N7x9MpZIKDlZzN+vg="; + }; + + build-system = [ + setuptools + ]; + + dependencies = [ + pyvers + torch + ]; + + pythonImportsCheck = [ "hoptorch" ]; + + checkInputs = lib.optionals stdenv.hostPlatform.isDarwin [ + # torch._inductor.exc.InductorError: CppCompileError: C++ compile error + # fatal error: 'omp.h' file not found + llvmPackages.openmp + ]; + + nativeCheckInputs = [ + pytestCheckHook + ]; + + meta = { + description = "Small compatibility package for PyTorch higher-order operators"; + homepage = "https://github.com/vmoens/hoptorch"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ GaetanLepage ]; + }; +}) diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix index fe0b5c0e2abe..16dc8a511930 100644 --- a/pkgs/development/python-modules/tensordict/default.nix +++ b/pkgs/development/python-modules/tensordict/default.nix @@ -29,7 +29,7 @@ buildPythonPackage (finalAttrs: { pname = "tensordict"; - version = "0.12.4"; + version = "0.13.0"; pyproject = true; __structuredAttrs = true; @@ -37,7 +37,7 @@ buildPythonPackage (finalAttrs: { owner = "pytorch"; repo = "tensordict"; tag = "v${finalAttrs.version}"; - hash = "sha256-P5Mw3Hi+zOQBdE2O2gk7VI0/y58v0ogmL3Fj1996Gdg="; + hash = "sha256-JL6S3Wc9PJbskOPhsr+0QFunZBkMcJCsvhLlz6ggAQ4="; }; postPatch = '' diff --git a/pkgs/development/python-modules/torchrl/default.nix b/pkgs/development/python-modules/torchrl/default.nix index fdc32563e15e..7a8f43834926 100644 --- a/pkgs/development/python-modules/torchrl/default.nix +++ b/pkgs/development/python-modules/torchrl/default.nix @@ -75,14 +75,15 @@ buildPythonPackage (finalAttrs: { pname = "torchrl"; - version = "0.11.1"; + version = "0.13.1"; pyproject = true; + __structuredAttrs = true; src = fetchFromGitHub { owner = "pytorch"; repo = "rl"; tag = "v${finalAttrs.version}"; - hash = "sha256-1S/A9zb6hlRYIV8Jf1lQ31TzxuA16lDiQHTu+Y6WSQk="; + hash = "sha256-S0jfngqmRQwIVGHM72KUnp/TlR2js+QLRpsX6abiv84="; }; postPatch = '' diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 13f7492df9cb..0b679e3516fd 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -7246,6 +7246,8 @@ self: super: with self; { hopcroftkarp = callPackage ../development/python-modules/hopcroftkarp { }; + hoptorch = callPackage ../development/python-modules/hoptorch { }; + horimote = callPackage ../development/python-modules/horimote { }; horizon-eda = callPackage ../development/python-modules/horizon-eda { inherit (pkgs) horizon-eda; };