depot/third_party/nixpkgs/pkgs/development/python-modules/duckdb/default.nix
Default email 5e9e1146e1 Project import generated by Copybara.
GitOrigin-RevId: 18036c0be90f4e308ae3ebcab0e14aae0336fe42
2023-08-05 00:07:22 +02:00

68 lines
1.3 KiB
Nix

{ lib
, buildPythonPackage
, duckdb
, google-cloud-storage
, numpy
, pandas
, psutil
, pybind11
, setuptools-scm
, pytestCheckHook
}:
buildPythonPackage rec {
inherit (duckdb) pname version src patches;
format = "setuptools";
postPatch = ''
# we can't use sourceRoot otherwise patches don't apply, because the patches apply to the C++ library
cd tools/pythonpkg
# 1. let nix control build cores
# 2. unconstrain setuptools_scm version
substituteInPlace setup.py \
--replace "multiprocessing.cpu_count()" "$NIX_BUILD_CORES" \
--replace "setuptools_scm<7.0.0" "setuptools_scm"
# avoid dependency on mypy
rm tests/stubs/test_stubs.py
'';
SETUPTOOLS_SCM_PRETEND_VERSION = version;
nativeBuildInputs = [
pybind11
setuptools-scm
];
propagatedBuildInputs = [
numpy
pandas
];
nativeCheckInputs = [
google-cloud-storage
psutil
pytestCheckHook
];
disabledTests = [
# tries to make http request
"test_install_non_existent_extension"
];
preCheck = ''
export HOME="$(mktemp -d)"
'';
pythonImportsCheck = [
"duckdb"
];
meta = with lib; {
description = "Python binding for DuckDB";
homepage = "https://duckdb.org/";
license = licenses.mit;
maintainers = with maintainers; [ cpcloud ];
};
}