mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-11-17 19:21:04 +00:00
python312Packages.ignite: refactor (#334523)
This commit is contained in:
commit
a12d474186
|
@ -2,22 +2,23 @@
|
|||
lib,
|
||||
buildPythonPackage,
|
||||
fetchFromGitHub,
|
||||
pythonOlder,
|
||||
setuptools,
|
||||
pytestCheckHook,
|
||||
pytest-xdist,
|
||||
torchvision,
|
||||
pythonOlder,
|
||||
matplotlib,
|
||||
mock,
|
||||
packaging,
|
||||
torch,
|
||||
scikit-learn,
|
||||
tqdm,
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "ignite";
|
||||
version = "0.5.1";
|
||||
format = "setuptools";
|
||||
pyproject = true;
|
||||
|
||||
disabled = pythonOlder "3.8";
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "pytorch";
|
||||
|
@ -26,6 +27,13 @@ buildPythonPackage rec {
|
|||
hash = "sha256-J0xrqAGwH0bAs59T7zA8irMWOGbE2+Zd9kwqxYUYYMA=";
|
||||
};
|
||||
|
||||
build-system = [ setuptools ];
|
||||
|
||||
dependencies = [
|
||||
packaging
|
||||
torch
|
||||
];
|
||||
|
||||
nativeCheckInputs = [
|
||||
pytestCheckHook
|
||||
matplotlib
|
||||
|
@ -33,12 +41,6 @@ buildPythonPackage rec {
|
|||
pytest-xdist
|
||||
torchvision
|
||||
];
|
||||
propagatedBuildInputs = [
|
||||
packaging
|
||||
torch
|
||||
scikit-learn
|
||||
tqdm
|
||||
];
|
||||
|
||||
# runs successfully in 3.9, however, async isn't correctly closed so it will fail after test suite.
|
||||
doCheck = pythonOlder "3.9";
|
||||
|
@ -76,10 +78,22 @@ buildPythonPackage rec {
|
|||
"visdom"
|
||||
];
|
||||
|
||||
meta = with lib; {
|
||||
pythonImportsCheck = [
|
||||
"ignite"
|
||||
"ignite.engine"
|
||||
"ignite.handlers"
|
||||
"ignite.metrics"
|
||||
"ignite.distributed"
|
||||
"ignite.exceptions"
|
||||
"ignite.utils"
|
||||
"ignite.contrib"
|
||||
];
|
||||
|
||||
meta = {
|
||||
description = "High-level training library for PyTorch";
|
||||
homepage = "https://pytorch.org/ignite";
|
||||
license = licenses.bsd3;
|
||||
maintainers = [ maintainers.bcdarwin ];
|
||||
homepage = "https://pytorch-ignite.ai";
|
||||
changelog = "https://github.com/pytorch/ignite/releases/tag/v${version}";
|
||||
license = lib.licenses.bsd3;
|
||||
maintainers = [ lib.maintainers.bcdarwin ];
|
||||
};
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue