Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.8 kB · 126 lines
Nix
at litex
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127{ lib, autograd, buildPythonPackage, cupy, cvxopt, cython, fetchPypi, matplotlib, numpy, tensorflow, pymanopt, pytestCheckHook, pythonOlder, scikit-learn, scipy, enableDimensionalityReduction ? false, enableGPU ? false}:
buildPythonPackage rec { pname = "pot"; version = "0.8.2"; format = "setuptools";
disabled = pythonOlder "3.6";
src = fetchPypi { pname = "POT"; inherit version; hash = "sha256-PKmuPI83DPy7RkOgHHPdPJJz5NT/fpr123AVTzTLwgQ="; };
nativeBuildInputs = [ numpy cython ];
propagatedBuildInputs = [ numpy scipy ] ++ lib.optionals enableGPU [ cupy ] ++ lib.optionals enableDimensionalityReduction [ autograd pymanopt ];
nativeCheckInputs = [ cvxopt matplotlib numpy tensorflow scikit-learn pytestCheckHook ];
postPatch = '' substituteInPlace setup.cfg \ --replace " --cov-report= --cov=ot" "" \ --replace " --durations=20" "" \ --replace " --junit-xml=junit-results.xml" "" substituteInPlace setup.py \ --replace '"oldest-supported-numpy", ' ""
# we don't need setup.py to find the macos sdk for us sed -i '/sdk_path/d' setup.py '';
# To prevent importing of an incomplete package from the build directory # instead of nix store (`ot` is the top-level package name). preCheck = '' rm -r ot '';
disabledTests = [ # GPU tests are always skipped because of sandboxing "warnings" # Fixture is not available "test_conditional_gradient" "test_convert_between_backends" "test_emd_backends" "test_emd_emd2_types_devices" "test_emd1d_type_devices" "test_emd2_backends" "test_factored_ot_backends" "test_free_support_barycenter_backends" "test_func_backends" "test_generalized_conditional_gradient" "test_line_search_armijo" "test_loss_dual" "test_max_sliced_backend" "test_plan_dual" "test_random_backends" "test_sliced_backend" "test_to_numpy" "test_wasserstein_1d_type_devices" "test_wasserstein" "test_weak_ot_bakends" # TypeError: Only integers, slices... "test_emd1d_device_tf" ];
disabledTestPaths = [ # AttributeError: module pytest has no attribute skip_backend "test/test_bregman.py" "test/test_da.py" "test/test_utils.py" "test/test_gromov.py" "test/test_helpers.py" "test/test_unbalanced.py" ] ++ lib.optionals (!enableDimensionalityReduction) [ "test/test_dr.py" ];
pythonImportsCheck = [ "ot" "ot.lp" ];
meta = with lib; { description = "Python Optimal Transport Library"; homepage = "https://pythonot.github.io/"; license = licenses.mit; maintainers = with maintainers; [ yl3dy ]; };}