From ba43d90b78db469a028763eeac38c931d238faee Mon Sep 17 00:00:00 2001
From: Tom Sydney Kerckhove <NorfairKing@users.noreply.github.com>
Date: Sat, 30 Jan 2021 19:56:39 +0100
Subject: [PATCH] torch fix, fixes #230 (#237)

* torch fix, fixes #230

Co-authored-by: Tom Sydney Kerckhove <syd@cs-syd.eu>
---
 overrides.nix | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/overrides.nix b/overrides.nix
index 5f1c619..f3cbfbe 100644
--- a/overrides.nix
+++ b/overrides.nix
@@ -1184,14 +1184,16 @@ self: super:
           # is explicitly disabled with USE_CUDA=0.
           find $out -name "*.so" -exec ${pkgs.patchelf}/bin/patchelf --remove-needed libcuda.so.1 {} \;
         '';
-        buildInputs = old.buildInputs ++ lib.optionals enableCuda [
+        buildInputs = (old.buildInputs or [ ])
+          ++ [ self.typing-extensions ]
+          ++ lib.optionals enableCuda [
           pkgs.linuxPackages.nvidia_x11
           pkgs.nccl.dev
           pkgs.nccl.out
         ];
         propagatedBuildInputs = [
-          super.numpy
-          super.future
+          self.numpy
+          self.future
         ];
       })
     )