3
0
Fork 0
forked from mirrors/nixpkgs
nixpkgs/pkgs/applications/version-management/dvc/default.nix
2022-06-02 15:37:20 +02:00

128 lines
2.6 KiB
Nix

{ lib
, python3
, fetchFromGitHub
, fetchpatch
, enableGoogle ? false
, enableAWS ? false
, enableAzure ? false
, enableSSH ? false
}:
let
py = python3.override {
packageOverrides = self: super: {
grandalf = super.grandalf.overridePythonAttrs (oldAttrs: rec {
version = "0.6";
src = fetchFromGitHub {
owner = "bdcht";
repo = "grandalf";
rev = "v${version}";
hash = "sha256-T4pVzjz1WbfBA2ybN4IRK73PD/eb83YUW0BZrBESNLg=";
};
postPatch = ''
substituteInPlace setup.py \
--replace "setup_requires=['pytest-runner',]," ""
'';
});
scmrepo = super.scmrepo.overridePythonAttrs (oldAttrs: rec {
version = "0.0.19";
src = fetchFromGitHub {
owner = "iterative";
repo = "scmrepo";
rev = "refs/tags/${version}";
hash = "sha256-f/KV3NfIumkZcg9r421QhdyPU/274aAU4b78myi+fFY=";
};
});
};
};
in
with py.pkgs;
buildPythonApplication rec {
pname = "dvc";
version = "2.10.2";
format = "setuptools";
src = fetchFromGitHub {
owner = "iterative";
repo = pname;
rev = version;
hash = "sha256-boaQSg0jajWQZKB5wvcP2musVR2/pifT4pU64Y5hiQ0=";
};
nativeBuildInputs = with py.pkgs; [
setuptools-scm
setuptools-scm-git-archive
];
propagatedBuildInputs = with py.pkgs; [
aiohttp-retry
appdirs
colorama
configobj
configobj
dictdiffer
diskcache
distro
dpath
dvclive
dvc-render
flatten-dict
flufl_lock
funcy
grandalf
nanotime
networkx
packaging
pathspec
ply
psutil
pydot
pygtrie
pyparsing
python-benedict
requests
rich
ruamel-yaml
scmrepo
shortuuid
shtab
tabulate
toml
tqdm
typing-extensions
voluptuous
zc_lockfile
] ++ lib.optional enableGoogle [
google-cloud-storage
] ++ lib.optional enableAWS [
boto3
] ++ lib.optional enableAzure [
azure-storage-blob
] ++ lib.optional enableSSH [
paramiko
] ++ lib.optionals (pythonOlder "3.8") [
importlib-metadata
] ++ lib.optionals (pythonOlder "3.9") [
importlib-resources
];
postPatch = ''
substituteInPlace dvc/daemon.py \
--subst-var-by dvc "$out/bin/dcv"
'';
# Tests require access to real cloud services
doCheck = false;
meta = with lib; {
description = "Version Control System for Machine Learning Projects";
homepage = "https://dvc.org";
license = licenses.asl20;
maintainers = with maintainers; [ cmcdragonkai fab ];
};
}