nixpkgs/pkgs/development/python-modules/vowpalwabbit/default.nix

76 lines
1.6 KiB
Nix

{ stdenv
, lib
, fetchPypi
, buildPythonPackage
, cmake
, python
, zlib
, ncurses
, docutils
, pygments
, numpy
, scipy
, scikit-learn
, spdlog
, fmt
, rapidjson
}:
buildPythonPackage rec {
pname = "vowpalwabbit";
version = "9.0.1";
src = fetchPypi{
inherit pname version;
sha256 = "sha256-rDUgOjroY8S73+s+MWyBl86u+8XCH4O8KMCo9zUlqAs=";
};
nativeBuildInputs = [
cmake
];
buildInputs = [
docutils
ncurses
pygments
python.pkgs.boost
zlib.dev
spdlog
fmt
rapidjson
];
# As we disable configure via cmake, pass explicit global options to enable
# spdlog and fmt packages
setupPyGlobalFlags = [ "--cmake-options=\"-DSPDLOG_SYS_DEP=ON;-DFMT_SYS_DEP=ON\"" ];
propagatedBuildInputs = [
numpy
scikit-learn
scipy
];
# Python build script uses CMake, but we don't want CMake to do the
# configuration.
dontUseCmakeConfigure = true;
# Python ctypes.find_library uses DYLD_LIBRARY_PATH.
preConfigure = lib.optionalString stdenv.isDarwin ''
export DYLD_LIBRARY_PATH="${python.pkgs.boost}/lib"
'';
checkPhase = ''
# check-manifest requires a git clone, not a tarball
# check-manifest --ignore "Makefile,PACKAGE.rst,*.cc,tox.ini,tests*,examples*,src*"
${python.interpreter} setup.py check -ms
'';
meta = with lib; {
description = "Vowpal Wabbit is a fast machine learning library for online learning, and this is the python wrapper for the project.";
homepage = "https://github.com/JohnLangford/vowpal_wabbit";
license = licenses.bsd3;
broken = stdenv.isAarch64;
maintainers = with maintainers; [ teh ];
};
}