Luess/flake.nix at main · JakobFanselow/Luess · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
{
  description = "Python development environment";

  inputs.nixpkgs.url = "github:NixOS/nixpkgs/";

  outputs = { self, nixpkgs }:
    let
      supportedSystems = [ "x86_64-linux" "aarch64-linux" "x86_64-darwin" "aarch64-darwin" ];
      forEachSupportedSystem = f: nixpkgs.lib.genAttrs supportedSystems (system: f {
        pkgs = import nixpkgs { inherit system; };
      });

    in
    {
      devShells = forEachSupportedSystem ({ pkgs }: {
        default = pkgs.mkShell.override
          {

          }
          {

            buildInputs = with pkgs; [
              stdenv
              python3
              kaggle
              stockfish
            ];
            nativeBuildInputs = with pkgs; [
              libgcc
              pkg-config
              stdenv
            ];

            packages = with pkgs; [
		            python3Packages.jupyter
                python3Packages.ipython
                python3Packages.notebook
                python3Packages.numpy
                python3Packages.pandas
                python3Packages.matplotlib
                python3Packages.scipy
                #python3Packages.torch
                #python3Packages.torchvision
                python3Packages.tqdm
                python3Packages.torchmetrics
                python3Packages.mlxtend
                python3Packages.scikit-learn
                python3Packages.chess
                python3Packages.kagglehub
                python3Packages.h5py
                python3Packages.wandb
                python3Packages.requests
                python3Packages.zstandard
            ] ++ (if stdenv.hostPlatform.system == "aarch64-darwin" then [ ] else [

            ]);

            shellHook = ''
              python -m venv .venv
              source .venv/bin/activate
              pip install stockfish
              pip install torch
            '';
          };
      });


    };
}