Something went wrong. Try again.
lol
Something went wrong. Try again.
2.0 kB · 83 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384{ lib, stdenv, buildPythonPackage, fetchFromGitHub,
# build-system setuptools, setuptools-scm,
# dependencies linear-operator, mpmath, scikit-learn, scipy, torch,
# tests pytestCheckHook,}:
buildPythonPackage (finalAttrs: { pname = "gpytorch"; version = "1.15.2"; pyproject = true;
src = fetchFromGitHub { owner = "cornellius-gp"; repo = "gpytorch"; tag = "v${finalAttrs.version}"; hash = "sha256-1CavS+qrV8YqnsT87GjmJV2LOtvExFYQE5YpYZEw9ts="; };
build-system = [ setuptools setuptools-scm ];
dependencies = [ linear-operator mpmath scikit-learn scipy torch ];
nativeCheckInputs = [ pytestCheckHook ];
pythonImportsCheck = [ "gpytorch" ];
disabledTests = [ # AssertionError on number of warnings emitted "test_deprecated_methods" # flaky numerical tests "test_classification_error" "test_matmul_matrix_broadcast" "test_optimization_optimal_error" # https://github.com/cornellius-gp/gpytorch/issues/2396 "test_t_matmul_matrix" ] ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ # RuntimeError: Failed to initialize cpuinfo! "test_dtype_value_context" "test_half" ];
disabledTestPaths = lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [ # Hang forever "test/examples/test_spectral_mixture_gp_regression.py" "test/kernels/test_spectral_mixture_kernel.py" "test/utils/test_nearest_neighbors.py" "test/variational/test_nearest_neighbor_variational_strategy.py" ];
meta = { description = "Highly efficient and modular implementation of Gaussian Processes, with GPU acceleration"; homepage = "https://gpytorch.ai"; downloadPage = "https://github.com/cornellius-gp/gpytorch"; changelog = "https://github.com/cornellius-gp/gpytorch/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ veprbl ]; };})