-
Notifications
You must be signed in to change notification settings - Fork 0
/
flake.nix
84 lines (77 loc) · 1.87 KB
/
flake.nix
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
{
description = "";
inputs = {
nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
flake-utils.url = "github:numtide/flake-utils";
gitignore.url = "github:hercules-ci/gitignore.nix";
tinygrad.url = "github:wozeparrot/tinygrad-nix";
};
outputs = inputs @ {
nixpkgs,
flake-utils,
gitignore,
...
}: let
inherit (gitignore.lib) gitignoreSource;
overlay = final: prev: {
pythonPackagesExtensions =
prev.pythonPackagesExtensions
++ [
(
python-final: python-prev: {
tgim = python-prev.buildPythonPackage {
pname = "tgim";
version = "0.0.0";
pyproject = true;
src = gitignoreSource ./.;
nativeBuildInputs = with python-prev; [
setuptools
wheel
];
propagatedBuildInputs = with python-final; [
tinygrad
];
doCheck = false;
};
}
)
];
};
in
flake-utils.lib.eachDefaultSystem
(
system: let
pkgs = import nixpkgs {
inherit system;
overlays = [
inputs.tinygrad.overlays.default
overlay
];
};
in {
packages = rec {
inherit (pkgs.python3Packages) tgim;
default = tgim;
};
devShell = pkgs.mkShell {
packages = let
python-packages = p:
with p; [
pillow
tinygrad
numpy
setuptools
build
twine
];
python = pkgs.python311;
in [
(python.withPackages python-packages)
];
};
}
)
// {
overlays.default = overlay;
};
}