depot/third_party/nixpkgs/pkgs/development/python-modules/papermill/default.nix
Default email c7cb07f092 Project import generated by Copybara.
GitOrigin-RevId: 1536926ef5621b09bba54035ae2bb6d806d72ac8
2024-02-29 21:09:43 +01:00

108 lines
1.8 KiB
Nix

{ lib
, stdenv
, azure-datalake-store
, azure-identity
, azure-storage-blob
, boto3
, buildPythonPackage
, click
, entrypoints
, fetchFromGitHub
, gcsfs
, ipykernel
, moto
, nbclient
, nbformat
, pyarrow
, pygithub
, pytest-mock
, pytestCheckHook
, pythonOlder
, pyyaml
, requests
, setuptools
, tenacity
, tqdm
}:
buildPythonPackage rec {
pname = "papermill";
version = "2.5.0";
pyproject = true;
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = "nteract";
repo = "papermill";
rev = "refs/tags/${version}";
hash = "sha256-x6f5hhTdOPDVFiBvRhfrXq1wd5keYiuUshXnT0IkjX0=";
};
nativeBuildInputs = [
setuptools
];
propagatedBuildInputs = [
click
pyyaml
nbformat
nbclient
tqdm
requests
entrypoints
tenacity
];
passthru.optional-dependencies = {
azure = [
azure-datalake-store
azure-identity
azure-storage-blob
];
gcs = [
gcsfs
];
github = [
pygithub
];
hdfs = [
pyarrow
];
s3 = [
boto3
];
};
nativeCheckInputs = [
ipykernel
moto
pytest-mock
pytestCheckHook
] ++ passthru.optional-dependencies.azure
++ passthru.optional-dependencies.s3
++ passthru.optional-dependencies.gcs;
preCheck = ''
export HOME=$(mktemp -d)
'';
pythonImportsCheck = [
"papermill"
];
disabledTests = lib.optionals stdenv.isDarwin [
# might fail due to the sandbox
"test_end2end_autosave_slow_notebook"
];
__darwinAllowLocalNetworking = true;
meta = with lib; {
description = "Parametrize and run Jupyter and interact with notebooks";
homepage = "https://github.com/nteract/papermill";
license = licenses.bsd3;
maintainers = with maintainers; [ ];
mainProgram = "papermill";
};
}