depot/third_party/nixpkgs/pkgs/development/python-modules/mlrose/default.nix

{ lib
, isPy27
, buildPythonPackage
, fetchFromGitHub
, fetchpatch
, scikit-learn
, pytestCheckHook
, pytest-randomly
}:

buildPythonPackage rec {
  pname = "mlrose";
  version = "1.3.0";
  format = "setuptools";
  disabled = isPy27;

  src = fetchFromGitHub {
    owner = "gkhayes";
    repo = "mlrose";
    rev = "v${version}";
    sha256 = "1dn43k3rcypj58ymcj849b37w66jz7fphw8842v6mlbij3x0rxfl";
  };

  patches = [
    # Fixes compatibility with scikit-learn 0.24.1
    (fetchpatch {
      url = "https://github.com/gkhayes/mlrose/pull/55/commits/19caf8616fc194402678aa67917db334ad02852a.patch";
      sha256 = "1nivz3bn21nd21bxbcl16a6jmy7y5j8ilz90cjmd0xq4v7flsahf";
    })
  ];

  propagatedBuildInputs = [ scikit-learn ];
  nativeCheckInputs = [ pytest-randomly pytestCheckHook ];

  postPatch = ''
    substituteInPlace setup.py --replace sklearn scikit-learn
  '';

  pythonImportsCheck = [ "mlrose" ];

  # Fix random seed during tests
  pytestFlagsArray = [ "--randomly-seed 0" ];

  meta = with lib; {
    description = "Machine Learning, Randomized Optimization and SEarch";
    homepage    = "https://github.com/gkhayes/mlrose";
    license     = licenses.bsd3;
    maintainers = with maintainers; [ abbradar ];
  };
}