Testing ORT 1.15.0-rc1

oracle · May 8, 2023 · 8969ae2 · 8969ae2
1 parent 3ba47e6
commit 8969ae2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/Interop/ONNX/pom.xml b/Interop/ONNX/pom.xml
@@ -28,6 +28,20 @@
     <artifactId>tribuo-onnx</artifactId>
     <packaging>jar</packaging>
 
+    <!-- staging repo for ORT release candidates -->
+    <repositories>
+        <repository>
+            <id>ORT-Nightly</id>
+            <url>https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/maven/v1</url>
+            <releases>
+                <enabled>true</enabled>
+            </releases>
+            <snapshots>
+                <enabled>true</enabled>
+            </snapshots>
+        </repository>
+    </repositories>
+
     <dependencies>
         <dependency>
             <groupId>${project.groupId}</groupId>

diff --git a/Interop/ONNX/src/main/java/org/tribuo/interop/onnx/LabelTransformer.java b/Interop/ONNX/src/main/java/org/tribuo/interop/onnx/LabelTransformer.java
@@ -17,6 +17,7 @@
 package org.tribuo.interop.onnx;
 
 import ai.onnxruntime.OnnxJavaType;
+import ai.onnxruntime.OnnxMap;
 import ai.onnxruntime.OnnxSequence;
 import ai.onnxruntime.OnnxTensor;
 import ai.onnxruntime.OnnxValue;
@@ -184,8 +185,9 @@ protected float[][] getBatchPredictions(List<OnnxValue> inputs, ImmutableOutputI
                         float[][] outputArray = new float[output.size()][outputIDInfo.size()];
                         int i = 0;
                         for (Object o : output) {
+                            OnnxMap onnxMap = (OnnxMap) o;
                             @SuppressWarnings("unchecked") // guarded by the if on the mapInfo above.
-                            Map<Long,Float> map = (Map<Long,Float>) o;
+                            Map<Long,Float> map = (Map<Long,Float>) onnxMap.getValue();
                             if (map.size() == outputIDInfo.size()) {
                                 for (Map.Entry<Long,Float> e : map.entrySet()) {
                                     Long key = e.getKey();

diff --git a/pom.xml b/pom.xml
@@ -51,7 +51,7 @@
         <!-- 3rd party backend dependencies -->
         <liblinear.version>2.44</liblinear.version>
         <libsvm.version>3.25</libsvm.version>
-        <onnxruntime.version>1.12.1</onnxruntime.version>
+        <onnxruntime.version>1.15.0-rc1</onnxruntime.version>
         <tensorflow.version>0.5.0</tensorflow.version>
         <xgboost.version>1.6.2</xgboost.version>