58 lines
1.2 KiB
Nix
58 lines
1.2 KiB
Nix
|
{
|
||
|
|
lib,
|
||
|
|
python3,
|
||
|
|
fetchFromGitHub,
|
||
|
|
}:
|
||
|
|
|
||
|
|
python3.pkgs.buildPythonApplication rec {
|
||
|
|
pname = "modelscan";
|
||
|
|
version = "0.8.1";
|
||
|
|
pyproject = true;
|
||
|
|
|
||
|
|
src = fetchFromGitHub {
|
||
|
|
owner = "protectai";
|
||
|
|
repo = "modelscan";
|
||
|
|
rev = "refs/tags/v${version}";
|
||
|
|
hash = "sha256-90VnIVQFjtKgLCHc+tmOtDdgJP8aaH4h5ZiOOejnXgQ=";
|
||
|
|
};
|
||
|
|
|
||
|
|
build-system = with python3.pkgs; [
|
||
|
|
poetry-core
|
||
|
|
poetry-dynamic-versioning
|
||
|
|
];
|
||
|
|
|
||
|
|
dependencies = with python3.pkgs; [
|
||
|
|
click
|
||
|
|
numpy
|
||
|
|
rich
|
||
|
|
tomlkit
|
||
|
|
];
|
||
|
|
|
||
|
|
optional-dependencies = with python3.pkgs; {
|
||
|
|
h5py = [ h5py ];
|
||
|
|
# tensorflow = [ tensorflow ];
|
||
|
|
};
|
||
|
|
|
||
|
|
nativeCheckInputs =
|
||
|
|
with python3.pkgs;
|
||
|
|
[
|
||
|
|
dill
|
||
|
|
pytestCheckHook
|
||
|
|
]
|
||
|
|
++ lib.flatten (builtins.attrValues optional-dependencies);
|
||
|
|
|
||
|
|
# tensorflow doesn0t support Python 3.12
|
||
|
|
doCheck = false;
|
||
|
|
|
||
|
|
pythonImportsCheck = [ "modelscan" ];
|
||
|
|
|
||
|
|
meta = with lib; {
|
||
|
|
description = "Protection against Model Serialization Attacks";
|
||
|
|
homepage = "https://github.com/protectai/modelscan";
|
||
|
|
changelog = "https://github.com/protectai/modelscan/releases/tag/v${version}";
|
||
|
|
license = licenses.asl20;
|
||
|
|
maintainers = with maintainers; [ fab ];
|
||
|
|
mainProgram = "modelscan";
|
||
|
|
};
|
||
|
|
}
|