nixpkgs/pkgs/applications/misc/nimbo/default.nix

58 lines
1.3 KiB
Nix

{ lib
, python3
, fetchFromGitHub
, installShellFiles
, awscli
}:
python3.pkgs.buildPythonApplication rec {
pname = "nimbo";
version = "0.2.4";
disabled = python3.pythonOlder "3.6";
src = fetchFromGitHub {
owner = "nimbo-sh";
repo = pname;
rev = "v${version}";
sha256 = "1fs28s9ynfxrb4rzba6cmik0kl0q0vkpb4zdappsq62jqf960k24";
};
postPatch = ''
substituteInPlace setup.py \
--replace "awscli>=1.19<2.0" ""
'';
nativeBuildInputs = [ installShellFiles ];
propagatedBuildInputs = with python3.pkgs; [
setuptools
boto3
requests
click
pyyaml
pydantic
];
# nimbo tests require an AWS instance
doCheck = false;
pythonImportsCheck = [ "nimbo" ];
makeWrapperArgs = [
"--prefix" "PATH" ":" (lib.makeBinPath [ awscli ])
];
postInstall = ''
installShellCompletion --cmd nimbo \
--zsh <(_NIMBO_COMPLETE=source_zsh $out/bin/nimbo) \
--bash <(_NIMBO_COMPLETE=source_bash $out/bin/nimbo) \
--fish <(_NIMBO_COMPLETE=source_fish $out/bin/nimbo)
'';
meta = with lib; {
description = "Run machine learning jobs on AWS with a single command";
homepage = "https://github.com/nimbo-sh/nimbo";
license = licenses.bsl11;
maintainers = with maintainers; [ alexeyre noreferences ];
};
}