Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273{ lib, stdenv, buildPythonPackage, fetchFromGitHub,
# build-system setuptools, setuptools-scm,
# dependencies opt-einsum-fx, scipy, sympy, torch,
# tests pytestCheckHook, llvmPackages,}:
buildPythonPackage (finalAttrs: { pname = "e3nn"; version = "0.6.0"; pyproject = true; __structuredAttrs = true;
src = fetchFromGitHub { owner = "e3nn"; repo = "e3nn"; tag = finalAttrs.version; hash = "sha256-gGl0DiLU8w0jqGWA/ZzvkxdZdZCvtXqtmEEZ2dIwZ2o="; };
build-system = [ setuptools setuptools-scm ];
dependencies = [ opt-einsum-fx scipy sympy torch ];
pythonImportsCheck = [ "e3nn" ];
nativeCheckInputs = [ pytestCheckHook ];
disabledTests = [ # RuntimeError: torch.compile does not support compiling torch.jit.script or # torch.jit.freeze models directly "test_identity" ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # symbol not found in flat namespace '___kmpc_barrier' "test_activation" "test_input_weights_jit" "test_variance" ];
meta = { description = "Modular framework for neural networks with Euclidean symmetry"; homepage = "https://github.com/e3nn/e3nn"; changelog = "https://github.com/e3nn/e3nn/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; };})