99 lines
2.2 KiB
Nix
99 lines
2.2 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
buildPythonPackage,
|
|
pythonOlder,
|
|
rustPlatform,
|
|
cmake,
|
|
libiconv,
|
|
fetchFromGitHub,
|
|
typing-extensions,
|
|
jemalloc,
|
|
rust-jemalloc-sys,
|
|
darwin,
|
|
}:
|
|
|
|
let
|
|
rust-jemalloc-sys' = rust-jemalloc-sys.override {
|
|
jemalloc = jemalloc.override { disableInitExecTls = true; };
|
|
};
|
|
in
|
|
|
|
buildPythonPackage rec {
|
|
pname = "polars";
|
|
version = "0.20.15";
|
|
pyproject = true;
|
|
|
|
disabled = pythonOlder "3.8";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "pola-rs";
|
|
repo = "polars";
|
|
rev = "refs/tags/py-${version}";
|
|
hash = "sha256-N/VIi0s5unYWqlR5Mpaq9cqXl2ccbzWPuOtE2UbmQw8=";
|
|
};
|
|
|
|
# Cargo.lock file is sometimes behind actual release which throws an error,
|
|
# thus the `sed` command
|
|
# Make sure to check that the right substitutions are made when updating the package
|
|
preBuild = ''
|
|
#sed -i 's/version = "0.18.0"/version = "${version}"/g' Cargo.lock
|
|
'';
|
|
|
|
cargoDeps = rustPlatform.importCargoLock { lockFile = ./Cargo.lock; };
|
|
|
|
buildAndTestSubdir = "py-polars";
|
|
|
|
# Revisit this whenever package or Rust is upgraded
|
|
RUSTC_BOOTSTRAP = 1;
|
|
|
|
propagatedBuildInputs = lib.optionals (pythonOlder "3.11") [ typing-extensions ];
|
|
|
|
# trick taken from the polars repo since there seems to be a problem
|
|
# with simd enabled with our stable rust (instead of nightly).
|
|
maturinBuildFlags = [
|
|
"--no-default-features"
|
|
"--features=all"
|
|
];
|
|
|
|
dontUseCmakeConfigure = true;
|
|
|
|
nativeBuildInputs =
|
|
[
|
|
# needed for libz-ng-sys
|
|
# TODO: use pkgs.zlib-ng
|
|
cmake
|
|
]
|
|
++ (with rustPlatform; [
|
|
cargoSetupHook
|
|
maturinBuildHook
|
|
]);
|
|
|
|
buildInputs =
|
|
[ rust-jemalloc-sys' ]
|
|
++ lib.optionals stdenv.isDarwin [
|
|
libiconv
|
|
darwin.apple_sdk.frameworks.Security
|
|
darwin.apple_sdk.frameworks.SystemConfiguration
|
|
];
|
|
|
|
# nativeCheckInputs = [
|
|
# pytestCheckHook
|
|
# fixtures
|
|
# graphviz
|
|
# matplotlib
|
|
# networkx
|
|
# numpy
|
|
# pydot
|
|
# ];
|
|
|
|
pythonImportsCheck = [ "polars" ];
|
|
|
|
meta = with lib; {
|
|
description = "Fast multi-threaded DataFrame library";
|
|
homepage = "https://github.com/pola-rs/polars";
|
|
changelog = "https://github.com/pola-rs/polars/releases/tag/py-${version}";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ happysalada ];
|
|
};
|
|
}
|