Skip to content

Commit 3db6479

Browse files
authored
Merge pull request #91 from jturner314/patch-1
Fix compilation errors in tests
2 parents 5249646 + 8a035d7 commit 3db6479

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxruntime/tests/integration_tests.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ mod download {
103103
// and iterate on resulting probabilities, creating an index to later access labels.
104104
let mut probabilities: Vec<(usize, f32)> = outputs[0]
105105
.softmax(ndarray::Axis(1))
106-
.into_iter()
106+
.iter()
107107
.copied()
108108
.enumerate()
109109
.collect::<Vec<_>>();
@@ -192,7 +192,7 @@ mod download {
192192

193193
let mut probabilities: Vec<(usize, f32)> = outputs[0]
194194
.softmax(ndarray::Axis(1))
195-
.into_iter()
195+
.iter()
196196
.copied()
197197
.enumerate()
198198
.collect::<Vec<_>>();

0 commit comments

Comments
 (0)