95 lines
1.6 KiB
Nix
95 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 ];
|
|
};
|
|
}
|