Unverified Commit a12d4741 authored by Peder Bergebakken Sundt's avatar Peder Bergebakken Sundt Committed by GitHub
Browse files

python312Packages.ignite: refactor (#334523)

parents 98024df2 775776cd

pkgs/development/python-modules/ignite/default.nix

+28 −14
Original line number Diff line number Diff line
@@ -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 ];

  };

}