Something went wrong. Try again.
lol
Something went wrong. Try again.
2.3 kB · 110 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111{ lib, buildPythonPackage, fetchFromGitHub, fetchpatch,
# build-system setuptools,
# dependencies makefun, multipledispatch, numpy, opt-einsum, typing-extensions,
# checks pyro-ppl, torch, pandas, pillow, pyro-api, pytestCheckHook, pytest-xdist, requests, scipy, torchvision,
stdenv,}:
buildPythonPackage rec { pname = "funsor"; version = "0.4.6"; pyproject = true;
src = fetchFromGitHub { owner = "pyro-ppl"; repo = "funsor"; tag = version; hash = "sha256-Prj1saT0yoPAP8rDE0ipBEpR3QMk4PS12VSJlxc22p8="; };
patches = [ # Compatibility with torch >= 2.5 (arg_constraints is now a property) (fetchpatch { url = "https://github.com/pyro-ppl/funsor/commit/c5e2a48d73cad4e98058147af4090171272a44e5.patch"; hash = "sha256-sTR+hbJtS0Th5sIqlvB2bReEC0wnEbnB7gAiZKiqjAQ="; }) ];
# TypeError: clip() got an unexpected keyword argument 'a_max' postPatch = '' substituteInPlace funsor/jax/ops.py \ --replace-fail "a_max=" "max=" \ --replace-fail "a_min=" "min=" '';
build-system = [ setuptools ];
dependencies = [ makefun multipledispatch numpy opt-einsum typing-extensions ];
nativeCheckInputs = [ # Backend pyro-ppl torch
pandas pillow pyro-api pytestCheckHook pytest-xdist requests scipy torchvision ];
preCheck = '' export FUNSOR_BACKEND=torch '';
pythonImportsCheck = [ "funsor" ];
disabledTests = [ # `test_torch_save` got broken by the update of torch (2.3.1 -> 2.4.0): # FutureWarning: You are using `torch.load` with `weights_only=False`... # TODO: Try to re-enable this test at next release "test_torch_save" ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # Failures related to JIT # RuntimeError: required keyword attribute 'Subgraph' has the wrong type "test_local_param_ok" "test_plate_ok" ];
meta = { description = "Functional tensors for probabilistic programming"; homepage = "https://funsor.pyro.ai"; changelog = "https://github.com/pyro-ppl/funsor/releases/tag/${version}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ GaetanLepage ]; };}