Fix 'IntArrayRef' error (#15)

SciNim · Feb 18, 2021 · cba9cc5 · cba9cc5
1 parent a070625
commit cba9cc5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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
 # -----------------------------------------------------------------------