Unverified Commit a7914b92 authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

Merge pull request #212993 from bcdarwin/update-tensorly 

python310Packages.tensorly: 0.7.0 -> 0.8.0
parents 4133eaef e07bc3dc

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

+22 −13
Original line number Diff line number Diff line 
{ lib

{ stdenv

, lib

, buildPythonPackage

, fetchFromGitHub

, numpy

, pytestCheckHook

, pythonOlder

, scipy

, sparse

}:



buildPythonPackage rec {

  pname = "tensorly";

  version = "0.7.0";

  version = "0.8.0";

  format = "setuptools";



  disabled = pythonOlder "3.7";
@@ -18,29 +18,35 @@ buildPythonPackage rec { 
  src = fetchFromGitHub {

    owner = pname;

    repo = pname;

    rev = version;

    sha256 = "sha256-VcX3pCczZQUYZaD7xrrkOcj0QPJt28cYTwpZm5D/X3c=";

    rev = "refs/tags/${version}";

    hash = "sha256-6iZvUgsoYf8fDGEuAODgfr4jCkiJwaJXlQUAsaOF9JU=";

  };



  propagatedBuildInputs = [

    numpy

    scipy

    sparse

  ];



  nativeCheckInputs = [

    pytestCheckHook

  ];



  postPatch = ''

    # nose is not actually required for anything

    # (including testing with the minimal dependencies)

    substituteInPlace setup.py \

      --replace ", 'nose'" ""

  '';



  pythonImportsCheck = [

    "tensorly"

    "tensorly.base"

    "tensorly.cp_tensor"

    "tensorly.tucker_tensor"

    "tensorly.tt_tensor"

    "tensorly.tt_matrix"

    "tensorly.parafac2_tensor"

    "tensorly.tenalg"

    "tensorly.decomposition"

    "tensorly.regression"

    "tensorly.metrics"

    "tensorly.random"

    "tensorly.datasets"

    "tensorly.plugins"

    "tensorly.contrib"

  ];



  pytestFlagsArray = [
@@ -48,6 +54,8 @@ buildPythonPackage rec { 
  ];



  disabledTests = [

    # tries to download data:

    "test_kinetic"

    # AssertionError: Partial_SVD took too long, maybe full_matrices set wrongly

    "test_svd_time"

  ];
@@ -57,5 +65,6 @@ buildPythonPackage rec { 
    homepage = "https://tensorly.org/";

    license = licenses.bsd3;

    maintainers = with maintainers; [ bcdarwin ];

    broken = stdenv.isLinux && stdenv.isAarch64;  # test failures: test_TTOI and test_validate_tucker_rank

  };

}