Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 58 lines
Nix
at llvm-cleanup
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859{ lib, buildPythonPackage, fetchFromGitHub, linear-operator, scikit-learn, setuptools, setuptools-scm, wheel, torch, pytestCheckHook,}:
buildPythonPackage rec { pname = "gpytorch"; version = "1.11"; format = "pyproject";
src = fetchFromGitHub { owner = "cornellius-gp"; repo = pname; rev = "v${version}"; hash = "sha256-cpkfjx5G/4duL1Rr4nkHTHi03TDcYbcx3bKP2Ny7Ijo="; };
nativeBuildInputs = [ setuptools setuptools-scm wheel ];
propagatedBuildInputs = [ linear-operator scikit-learn torch ];
checkInputs = [ pytestCheckHook ];
pythonImportsCheck = [ "gpytorch" ];
disabledTests = [ # AssertionError on number of warnings emitted "test_deprecated_methods" # flaky numerical tests "test_classification_error" "test_matmul_matrix_broadcast" # https://github.com/cornellius-gp/gpytorch/issues/2396 "test_t_matmul_matrix" ];
meta = with lib; { description = "A highly efficient and modular implementation of Gaussian Processes, with GPU acceleration"; homepage = "https://gpytorch.ai"; license = licenses.mit; maintainers = with maintainers; [ veprbl ]; };}