From cba9cc541d6af8d4987e98e794d12ffb51629fdb Mon Sep 17 00:00:00 2001
From: machineko <42299606+machineko@users.noreply.github.com>
Date: Thu, 18 Feb 2021 13:33:57 +0100
Subject: [PATCH] Fix 'IntArrayRef' error (#15)

---
 flambeau/raw_bindings/neural_nets.nim | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/flambeau/raw_bindings/neural_nets.nim b/flambeau/raw_bindings/neural_nets.nim
index a06ecf8..e790590 100644
--- a/flambeau/raw_bindings/neural_nets.nim
+++ b/flambeau/raw_bindings/neural_nets.nim
@@ -8,7 +8,8 @@
 import
   ../cpp/std_cpp,
   ../libtorch,
-  ./tensors
+  ./tensors,
+  ./c10
 
 # (Almost) raw bindings to PyTorch Neural Networks
 # -----------------------------------------------------------------------