diff --git a/pkgs/development/python-modules/mlxtend/default.nix b/pkgs/development/python-modules/mlxtend/default.nix new file mode 100644 index 000000000000..2bfd12249101 --- /dev/null +++ b/pkgs/development/python-modules/mlxtend/default.nix @@ -0,0 +1,49 @@ +{ stdenv +, buildPythonPackage +, fetchFromGitHub +, isPy27 +, pytestCheckHook +, scipy +, numpy +, scikitlearn +, pandas +, matplotlib +, joblib +}: + +buildPythonPackage rec { + pname = "mlxtend"; + version = "0.17.3"; + disabled = isPy27; + + src = fetchFromGitHub { + owner = "rasbt"; + repo = pname; + rev = version; + sha256 = "1515wgmj5rhwpmky7apmmvys1630sfg534fai6559s13hp11pdcl"; + }; + + checkInputs = [ pytestCheckHook ]; + # image tests download files over the network + pytestFlagsArray = [ "-sv" "--ignore=mlxtend/image" ]; + # Fixed in master, but failing in release version + # see: https://github.com/rasbt/mlxtend/pull/721 + disabledTests = [ "test_variance_explained_ratio" ]; + + propagatedBuildInputs = [ + scipy + numpy + scikitlearn + pandas + matplotlib + joblib + ]; + + meta = with stdenv.lib; { + description = "A library of Python tools and extensions for data science"; + homepage = "https://github.com/rasbt/mlxtend"; + license= licenses.bsd3; + maintainers = with maintainers; [ evax ]; + platforms = platforms.unix; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index d209c34bdd8f..94e7ec416517 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -3754,6 +3754,8 @@ in { mlrose = callPackage ../development/python-modules/mlrose { }; + mlxtend = callPackage ../development/python-modules/mlxtend { }; + mmh3 = callPackage ../development/python-modules/mmh3 { }; mmpython = callPackage ../development/python-modules/mmpython { };