96 lines
1.6 KiB
Nix
96 lines
1.6 KiB
Nix
|
{
|
||
|
lib,
|
||
|
stdenv,
|
||
|
buildPythonPackage,
|
||
|
rustPlatform,
|
||
|
fetchFromGitHub,
|
||
|
darwin,
|
||
|
libiconv,
|
||
|
pkg-config,
|
||
|
protobuf,
|
||
|
numpy,
|
||
|
pyarrow,
|
||
|
ml-dtypes,
|
||
|
pandas,
|
||
|
pillow,
|
||
|
polars,
|
||
|
pytestCheckHook,
|
||
|
tqdm,
|
||
|
}:
|
||
|
|
||
|
buildPythonPackage rec {
|
||
|
pname = "pylance";
|
||
|
version = "0.13.0";
|
||
|
pyproject = true;
|
||
|
|
||
|
src = fetchFromGitHub {
|
||
|
owner = "lancedb";
|
||
|
repo = "lance";
|
||
|
rev = "refs/tags/v${version}";
|
||
|
hash = "sha256-gwSpdj3i68QBrIcuvjj/32CsRKYVh9dSf98qNLDpxpc=";
|
||
|
};
|
||
|
|
||
|
sourceRoot = "${src.name}/python";
|
||
|
|
||
|
cargoDeps = rustPlatform.importCargoLock { lockFile = ./Cargo.lock; };
|
||
|
|
||
|
postPatch = ''
|
||
|
ln -s ${./Cargo.lock} Cargo.lock
|
||
|
'';
|
||
|
|
||
|
nativeBuildInputs = [
|
||
|
pkg-config
|
||
|
rustPlatform.cargoSetupHook
|
||
|
];
|
||
|
|
||
|
build-system = [ rustPlatform.maturinBuildHook ];
|
||
|
|
||
|
buildInputs =
|
||
|
[
|
||
|
libiconv
|
||
|
protobuf
|
||
|
]
|
||
|
++ lib.optionals stdenv.isDarwin (
|
||
|
with darwin.apple_sdk.frameworks;
|
||
|
[
|
||
|
Security
|
||
|
SystemConfiguration
|
||
|
]
|
||
|
);
|
||
|
|
||
|
pythonRelaxDeps = [ "pyarrow" ];
|
||
|
|
||
|
dependencies = [
|
||
|
numpy
|
||
|
pyarrow
|
||
|
];
|
||
|
|
||
|
pythonImportsCheck = [ "lance" ];
|
||
|
|
||
|
nativeCheckInputs = [
|
||
|
ml-dtypes
|
||
|
pandas
|
||
|
pillow
|
||
|
polars
|
||
|
pytestCheckHook
|
||
|
tqdm
|
||
|
];
|
||
|
|
||
|
preCheck = ''
|
||
|
cd python/tests
|
||
|
'';
|
||
|
|
||
|
disabledTests = [
|
||
|
# Error during planning: Invalid function 'invert'.
|
||
|
"test_polar_scan"
|
||
|
"test_simple_predicates"
|
||
|
];
|
||
|
|
||
|
meta = {
|
||
|
description = "Python wrapper for Lance columnar format";
|
||
|
homepage = "https://github.com/lancedb/lance";
|
||
|
license = lib.licenses.asl20;
|
||
|
maintainers = with lib.maintainers; [ natsukium ];
|
||
|
};
|
||
|
}
|