Guillaume Girol
33afbf39f6
checkInputs used to be added to nativeBuildInputs. Now we have nativeCheckInputs to do that instead. Doing this treewide change allows to keep hashes identical to before the introduction of nativeCheckInputs.
110 lines
1.8 KiB
Nix
110 lines
1.8 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, fetchFromGitHub
|
|
, emcee
|
|
, matplotlib
|
|
, netcdf4
|
|
, numba
|
|
, numpy
|
|
, pandas
|
|
, pytest
|
|
, setuptools
|
|
, cloudpickle
|
|
, pytestCheckHook
|
|
, scipy
|
|
, packaging
|
|
, typing-extensions
|
|
, pythonOlder
|
|
, xarray
|
|
, xarray-einstats
|
|
, zarr
|
|
, ffmpeg
|
|
, h5py
|
|
, jaxlib
|
|
, torchvision
|
|
, jax
|
|
# , pymc3 (circular dependency)
|
|
, pyro-ppl
|
|
#, pystan (not packaged)
|
|
, numpyro
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "arviz";
|
|
version = "0.14.0";
|
|
format = "setuptools";
|
|
|
|
disabled = pythonOlder "3.7";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "arviz-devs";
|
|
repo = pname;
|
|
rev = "refs/tags/${version}";
|
|
hash = "sha256-YLNczcgVmcctNc620Ap9yQtQTwF1LREtL57JIWS/DKQ=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
matplotlib
|
|
netcdf4
|
|
numpy
|
|
packaging
|
|
pandas
|
|
scipy
|
|
setuptools
|
|
xarray
|
|
xarray-einstats
|
|
];
|
|
|
|
nativeCheckInputs = [
|
|
cloudpickle
|
|
emcee
|
|
ffmpeg
|
|
h5py
|
|
jax
|
|
jaxlib
|
|
numba
|
|
numpyro
|
|
# pymc3 (circular dependency)
|
|
pyro-ppl
|
|
# pystan (not packaged)
|
|
pytestCheckHook
|
|
torchvision
|
|
zarr
|
|
];
|
|
|
|
preCheck = ''
|
|
export HOME=$(mktemp -d);
|
|
'';
|
|
|
|
pytestFlagsArray = [
|
|
"arviz/tests/base_tests/"
|
|
];
|
|
|
|
disabledTestPaths = [
|
|
# Remove tests as dependency creates a circular dependency
|
|
"arviz/tests/external_tests/test_data_pymc.py"
|
|
];
|
|
|
|
disabledTests = [
|
|
# Tests require network access
|
|
"test_plot_separation"
|
|
"test_plot_trace_legend"
|
|
"test_cov"
|
|
# countourpy is not available at the moment
|
|
"test_plot_kde"
|
|
"test_plot_kde_2d"
|
|
"test_plot_pair"
|
|
];
|
|
|
|
pythonImportsCheck = [
|
|
"arviz"
|
|
];
|
|
|
|
meta = with lib; {
|
|
description = "Library for exploratory analysis of Bayesian models";
|
|
homepage = "https://arviz-devs.github.io/arviz/";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ omnipotententity ];
|
|
};
|
|
}
|