depot/third_party/nixpkgs/pkgs/development/python-modules/pandas/default.nix
Default email 5e7c2d6cef Project import generated by Copybara.
GitOrigin-RevId: f99e5f03cc0aa231ab5950a15ed02afec45ed51a
2023-10-09 21:29:22 +02:00

250 lines
4.8 KiB
Nix

{ lib
, stdenv
, buildPythonPackage
, fetchPypi
, pythonOlder
# build-system
, cython
, oldest-supported-numpy
, setuptools
, versioneer
, wheel
# propagates
, numpy
, python-dateutil
, pytz
, tzdata
# optionals
, beautifulsoup4
, bottleneck
, blosc2
, brotlipy
, fsspec
, gcsfs
, html5lib
, jinja2
, lxml
, matplotlib
, numba
, numexpr
, odfpy
, openpyxl
, psycopg2
, pyarrow
, pymysql
, pyqt5
, pyreadstat
, python-snappy
, qtpy
, s3fs
, scipy
, sqlalchemy
, tables
, tabulate
, xarray
, xlrd
, xlsxwriter
, zstandard
# tests
, adv_cmds
, glibc
, glibcLocales
, hypothesis
, pytestCheckHook
, pytest-xdist
, pytest-asyncio
, python
, runtimeShell
}:
buildPythonPackage rec {
pname = "pandas";
version = "2.0.3";
format = "pyproject";
disabled = pythonOlder "3.8";
src = fetchPypi {
inherit pname version;
hash = "sha256-wC83Kojg0X820wk6ZExzz8F4jodqfEvLQCCndRLiBDw=";
};
nativeBuildInputs = [
setuptools
cython
numpy
oldest-supported-numpy
versioneer
wheel
] ++ versioneer.optional-dependencies.toml;
enableParallelBuilding = true;
propagatedBuildInputs = [
numpy
python-dateutil
pytz
tzdata
];
passthru.optional-dependencies = let
extras = {
aws = [
s3fs
];
clipboard = [
pyqt5
qtpy
];
compression = [
brotlipy
python-snappy
zstandard
];
computation = [
scipy
xarray
];
excel = [
odfpy
openpyxl
# TODO: pyxlsb
xlrd
xlsxwriter
];
feather = [
pyarrow
];
fss = [
fsspec
];
gcp = [
gcsfs
# TODO: pandas-gqb
];
hdf5 = [
blosc2
tables
];
html = [
beautifulsoup4
html5lib
lxml
];
mysql = [
sqlalchemy
pymysql
];
output_formatting = [
jinja2
tabulate
];
parquet = [
pyarrow
];
performance = [
bottleneck
numba
numexpr
];
plot = [
matplotlib
];
postgresql = [
sqlalchemy
psycopg2
];
spss = [
pyreadstat
];
sql-other = [
sqlalchemy
];
xml = [
lxml
];
};
in extras // {
all = lib.concatLists (lib.attrValues extras);
};
nativeCheckInputs = [
glibcLocales
hypothesis
pytest-asyncio
pytest-xdist
pytestCheckHook
] ++ lib.optionals (stdenv.isLinux) [
# for locale executable
glibc
] ++ lib.optionals (stdenv.isDarwin) [
# for locale executable
adv_cmds
];
# don't max out build cores, it breaks tests
dontUsePytestXdist = true;
__darwinAllowLocalNetworking = true;
pytestFlagsArray = [
# https://github.com/pandas-dev/pandas/blob/main/test_fast.sh
"--skip-db"
"--skip-slow"
"--skip-network"
"-m" "'not single_cpu and not slow_arm'"
"--numprocesses" "4"
];
disabledTests = [
# AssertionError: Did not see expected warning of class 'FutureWarning'
"test_parsing_tzlocal_deprecated"
] ++ lib.optionals (stdenv.isDarwin && stdenv.isAarch64) [
# tests/generic/test_finalize.py::test_binops[and_-args4-right] - AssertionError: assert {} == {'a': 1}
"test_binops"
# These tests are unreliable on aarch64-darwin. See https://github.com/pandas-dev/pandas/issues/38921.
"test_rolling"
];
# Tests have relative paths, and need to reference compiled C extensions
# so change directory where `import .test` is able to be resolved
preCheck = ''
export HOME=$TMPDIR
export LC_ALL="en_US.UTF-8"
cd $out/${python.sitePackages}/pandas
''
# TODO: Get locale and clipboard support working on darwin.
# Until then we disable the tests.
+ lib.optionalString stdenv.isDarwin ''
# Fake the impure dependencies pbpaste and pbcopy
echo "#!${runtimeShell}" > pbcopy
echo "#!${runtimeShell}" > pbpaste
chmod a+x pbcopy pbpaste
export PATH=$(pwd):$PATH
'';
pythonImportsCheck = [
"pandas"
];
meta = with lib; {
# https://github.com/pandas-dev/pandas/issues/14866
# pandas devs are no longer testing i686 so safer to assume it's broken
broken = stdenv.isi686;
changelog = "https://pandas.pydata.org/docs/whatsnew/index.html";
description = "Powerful data structures for data analysis, time series, and statistics";
downloadPage = "https://github.com/pandas-dev/pandas";
homepage = "https://pandas.pydata.org";
license = licenses.bsd3;
longDescription = ''
Flexible and powerful data analysis / manipulation library for
Python, providing labeled data structures similar to R data.frame
objects, statistical functions, and much more.
'';
maintainers = with maintainers; [ raskin fridh knedlsepp ];
};
}