From e5317707f115fc8d6423e1d850c6b52d1f9edbb0 Mon Sep 17 00:00:00 2001
From: Christian Krinitsin <mail@krinitsin.com>
Date: Tue, 3 Jun 2025 12:02:16 +0000
Subject: fix shell.nix

---
 classification/shell.nix | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'classification')

diff --git a/classification/shell.nix b/classification/shell.nix
index bab1e7d2a..5a20b8e46 100644
--- a/classification/shell.nix
+++ b/classification/shell.nix
@@ -2,12 +2,13 @@
 with pkgs;
 mkShell {
   buildInputs = [
-    python312Packages.numpy
     python312Packages.transformers
     python312Packages.torch
   ];
   shellHook = ''
     # fixes libstdc++ issues and libgl.so issues
     LD_LIBRARY_PATH=${stdenv.cc.cc.lib}/lib/:/run/opengl-driver/lib/
+    # fixes unallocation issues
+    export MKL_NUM_THREADS=64
   '';
 }
-- 
cgit 1.4.1