default.nix

{ nixpkgs ? import <nixpkgs> {}
}:

let
  inherit (nixpkgs) pkgs;
  python = let
    packageOverrides = self: super: {
      # sklearn tests fail on darwin
      scikitlearn = super.scikitlearn.overridePythonAttrs(old: rec {
        doCheck = false;
      });
    };
  in pkgs.python3.override {inherit packageOverrides;};

  pythonPackages = with python.pkgs; [
    tensorflow
    matplotlib
    python
    scikitlearn
    Keras
  ];
in
  pkgs.stdenv.mkDerivation {
    name = "py-ml";
    buildInputs = pythonPackages;
  }