NixOSDots/dev-flakes/python/flake.nix

{
  description = "General Python dev environment";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
  };

  outputs = { self, nixpkgs, flake-utils }:
    flake-utils.lib.eachDefaultSystem (system:
      let
        pkgs = import nixpkgs {
          inherit system;
          config.allowUnfree = true;
        };

        python = pkgs.python311;
      in
      {
        devShells.default = pkgs.mkShell {
          name = "python-ml";

          packages = with pkgs; [
            python314
	    python314.pkgs.pygame
            #python.pkgs.pip
            #python.pkgs.virtualenv

            # ML stack
            #python.pkgs.numpy
            #python.pkgs.scipy
            #python.pkgs.pandas
            #python.pkgs.matplotlib
            #python.pkgs.scikit-learn

            # Optional
            #python.pkgs.jupyterlab
          ];

          shellHook = ''
            echo "Python loaded"
          '';
        };
      });
}