Unverified Commit 5ba549ea authored by Samuel Ainsworth's avatar Samuel Ainsworth Committed by GitHub
Browse files

Merge pull request #255709 from GaetanLepage/skorch 

python310Packages.skorch: 0.14.0 -> 0.15.0
parents 34424434 9ea632ea

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

+30 −8
Original line number Diff line number Diff line
@@ -2,28 +2,46 @@ 
, stdenv

, buildPythonPackage

, fetchPypi

, pytestCheckHook

, flaky

, pythonOlder

, numpy

, pandas

, torch

, scikit-learn

, scipy

, tabulate

, torch

, tqdm

, flaky

, pandas

, pytestCheckHook

, safetensors

, pythonAtLeast

}:



buildPythonPackage rec {

  pname = "skorch";

  version = "0.14.0";

  version = "0.15.0";



  src = fetchPypi {

    inherit pname version;

    hash = "sha256-/d0s0N40W18uGfVbD9VEbhbWfduoo+TBqDjmTkjMUxs=";

    hash = "sha256-39XVBlCmbg162z9uL84GZrU+v+M8waXbGdVV72ZYf84=";

  };



  propagatedBuildInputs = [ numpy torch scikit-learn scipy tabulate tqdm ];

  nativeCheckInputs = [ flaky pandas pytestCheckHook ];

  disabled = pythonOlder "3.8";



  propagatedBuildInputs = [

    numpy

    scikit-learn

    scipy

    tabulate

    torch

    tqdm

  ];



  nativeCheckInputs = [

    flaky

    pandas

    pytestCheckHook

    safetensors

  ];



  # patch out pytest-cov dep/invocation

  postPatch = ''
@@ -41,6 +59,10 @@ buildPythonPackage rec { 
  ] ++ lib.optionals stdenv.isDarwin [

    # there is a problem with the compiler selection

    "test_fit_and_predict_with_compile"

  ] ++ lib.optionals (pythonAtLeast "3.11") [

    # Python 3.11+ not yet supported for torch.compile

    # https://github.com/pytorch/pytorch/blob/v2.0.1/torch/_dynamo/eval_frame.py#L376-L377

    "test_fit_and_predict_with_compile"

  ];



  disabledTestPaths = [