nixpkgs/pkgs/development/python-modules/dvc/default.nix
2024-02-02 15:26:19 +01:00

175 lines
2.9 KiB
Nix

{ lib
, appdirs
, buildPythonPackage
, colorama
, configobj
, distro
, dpath
, dvc-azure
, dvc-data
, dvc-gdrive
, dvc-gs
, dvc-hdfs
, dvc-http
, dvc-render
, dvc-s3
, dvc-ssh
, dvc-studio-client
, dvc-task
, fetchFromGitHub
, flatten-dict
, flufl_lock
, funcy
, grandalf
, gto
, hydra-core
, importlib-metadata
, importlib-resources
, iterative-telemetry
, networkx
, packaging
, pathspec
, platformdirs
, psutil
, pydot
, pygtrie
, pyparsing
, pythonOlder
, pythonRelaxDepsHook
, requests
, rich
, ruamel-yaml
, scmrepo
, setuptools-scm
, shortuuid
, shtab
, tabulate
, tomlkit
, tqdm
, typing-extensions
, voluptuous
, zc-lockfile
, enableGoogle ? false
, enableAWS ? false
, enableAzure ? false
, enableSSH ? false
}:
buildPythonPackage rec {
pname = "dvc";
version = "3.43.1";
pyproject = true;
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = "iterative";
repo = "dvc";
rev = "refs/tags/${version}";
hash = "sha256-9JS8N4BWikmXoo5TtpUD5El2vHST80NEOBdxkHfK4ME=";
};
pythonRelaxDeps = [
"dvc-data"
"platformdirs"
];
postPatch = ''
substituteInPlace dvc/analytics.py \
--replace 'enabled = not os.getenv(DVC_NO_ANALYTICS)' 'enabled = False'
substituteInPlace dvc/daemon.py \
--subst-var-by dvc "$out/bin/dcv"
'';
nativeBuildInputs = [
pythonRelaxDepsHook
setuptools-scm
];
propagatedBuildInputs = [
appdirs
colorama
configobj
distro
dpath
dvc-data
dvc-http
dvc-render
dvc-studio-client
dvc-task
flatten-dict
flufl_lock
funcy
grandalf
gto
hydra-core
iterative-telemetry
networkx
packaging
pathspec
platformdirs
psutil
pydot
pygtrie
pyparsing
requests
rich
ruamel-yaml
scmrepo
shortuuid
shtab
tabulate
tomlkit
tqdm
typing-extensions
voluptuous
zc-lockfile
]
++ lib.optionals enableGoogle passthru.optional-dependencies.gs
++ lib.optionals enableAWS passthru.optional-dependencies.s3
++ lib.optionals enableAzure passthru.optional-dependencies.azure
++ lib.optionals enableSSH passthru.optional-dependencies.ssh
++ lib.optionals (pythonOlder "3.8") [
importlib-metadata
] ++ lib.optionals (pythonOlder "3.9") [
importlib-resources
];
passthru.optional-dependencies = {
azure = [
dvc-azure
];
gdrive = [
dvc-gdrive
];
gs = [
dvc-gs
];
hdfs = [
dvc-hdfs
];
s3 = [
dvc-s3
];
ssh = [
dvc-ssh
];
};
# Tests require access to real cloud services
doCheck = false;
pythonImportsCheck = [
"dvc"
"dvc.api"
];
meta = with lib; {
description = "Version Control System for Machine Learning Projects";
homepage = "https://dvc.org";
changelog = "https://github.com/iterative/dvc/releases/tag/${version}";
license = licenses.asl20;
maintainers = with maintainers; [ cmcdragonkai fab ];
};
}