Merge pull request apache#117 from ranjanan/ranjanan-depwarn

Get rid of `sub` depwarn
iblislin · Aug 22, 2016 · de77bce · de77bce
2 parents 8c22211 + 08ba2d1
commit de77bce
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/REQUIRE b/REQUIRE
@@ -1,5 +1,5 @@
 julia 0.4+
-Compat
+Compat 0.8.4
 Formatting
 BinDeps
 JSON
diff --git a/src/MXNet.jl b/src/MXNet.jl
@@ -9,6 +9,7 @@ module mx
 
 using Compat
 import Compat.String
+import Compat.view
 
 using Formatting
 

diff --git a/src/metric.jl b/src/metric.jl
@@ -69,7 +69,7 @@ function _update_single_output(metric :: Accuracy, label :: NDArray, pred :: NDA
         for j in 1:size(labels, 2)
           for i in 1:size(labels, 1)
             label = labels[i, j, 1, sample]
-            klasses = sub(pred, i, j, :, sample)
+            klasses = view(pred, i, j, :, sample)
             klass = indmax(klasses) - 1 # Classes start at 0...k-1
 
             metric.acc_sum += klass == label
@@ -79,7 +79,7 @@ function _update_single_output(metric :: Accuracy, label :: NDArray, pred :: NDA
       end
     elseif ndims(pred) == 2 # 1-dimensional case
       for sample in 1:size(label, 1)
-        klass = indmax(sub(pred, :, sample)) - 1
+        klass = indmax(view(pred, :, sample)) - 1
         metric.acc_sum += klass == label[sample]
         metric.n_sample += 1
       end