data loading and plotting for results wip

2025-09-03 14:55:54 +02:00
parent 3d968c305c
commit ed80faf1e2
16 changed files with 2732 additions and 952 deletions
--- a/tools/devenv.nix
+++ b/tools/devenv.nix
@@ -0,0 +1,27 @@
+{ pkgs, ... }:
+let
+  native_dependencies = with pkgs.python312Packages; [
+    torch-bin
+    torchvision-bin
+    aggdraw # for visualtorch
+    numpy
+    scipy
+    matplotlib
+  ];
+  tools = with pkgs; [
+    ruff
+  ];
+in
+{
+  packages = native_dependencies ++ tools;
+  languages.python = {
+    enable = true;
+    package = pkgs.python312;
+    uv = {
+      enable = true;
+      sync.enable = true;
+    };
+    venv.enable = true;
+  };
+
+}