change: upgrade to onnxruntime 1.21.0

Author: RJKeevil

Dockerfile

@@ -1,7 +1,7 @@
 #--- dockerfile to test hugot  ---
 
 ARG GO_VERSION=1.24.1
-ARG ONNXRUNTIME_VERSION=1.20.1
+ARG ONNXRUNTIME_VERSION=1.21.0
 ARG BUILD_PLATFORM=linux/amd64
 
 #--- runtime layer with all hugot dependencies for cpu ---

cuda.Dockerfile

@@ -1,7 +1,7 @@
 #--- dockerfile to test hugot  ---
 
 ARG GO_VERSION=1.24.1
-ARG ONNXRUNTIME_VERSION=1.20.1
+ARG ONNXRUNTIME_VERSION=1.21.0
 ARG BUILD_PLATFORM=linux/amd64
 
 #--- runtime layer with all hugot dependencies for cpu and gpu ---

go.mod

@@ -14,7 +14,7 @@ require (
 	github.com/stretchr/testify v1.10.0
 	github.com/urfave/cli/v2 v2.27.6
 	github.com/viant/afs v1.25.1
-	github.com/yalue/onnxruntime_go v1.17.0
+	github.com/yalue/onnxruntime_go v1.18.0
 	golang.org/x/exp v0.0.0-20250305212735-054e65f0b394
 )
 

go.sum

@@ -63,8 +63,8 @@ github.com/x448/float16 v0.8.4 h1:qLwI1I70+NjRFUR3zs1JPUCgaCXSh3SW62uAKT1mSBM=
 github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
 github.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 h1:gEOO8jv9F4OT7lGCjxCBTO/36wtF6j2nSip77qHd4x4=
 github.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1/go.mod h1:Ohn+xnUBiLI6FVj/9LpzZWtj1/D6lUovWYBkxHVV3aM=
-github.com/yalue/onnxruntime_go v1.17.0 h1:nC8AFbmaq9E2gxtxutGPzK/LGCrtnnu7LTGl82YuQzw=
-github.com/yalue/onnxruntime_go v1.17.0/go.mod h1:b4X26A8pekNb1ACJ58wAXgNKeUCGEAQ9dmACut9Sm/4=
+github.com/yalue/onnxruntime_go v1.18.0 h1:9/NZjgFeS4kOX8/kG5nMw1AxrLaREyvHRh7gTrred8Q=
+github.com/yalue/onnxruntime_go v1.18.0/go.mod h1:b4X26A8pekNb1ACJ58wAXgNKeUCGEAQ9dmACut9Sm/4=
 golang.org/x/crypto v0.36.0 h1:AnAEvhDddvBdpY+uR+MyHmuZzzNqXSe/GvuDeob5L34=
 golang.org/x/crypto v0.36.0/go.mod h1:Y4J0ReaxCR1IMaabaSMugxJES1EpwhBHhv2bDHklZvc=
 golang.org/x/exp v0.0.0-20250305212735-054e65f0b394 h1:nDVHiLt8aIbd/VzvPWN6kSOPE7+F/fNFDSXLVYkE/Iw=

hugot_training_test.go

@@ -15,6 +15,7 @@ import (
 	"github.com/knights-analytics/hugot/datasets"
 	"github.com/knights-analytics/hugot/options"
 	"github.com/knights-analytics/hugot/pipelines"
+	"github.com/knights-analytics/hugot/util"
 )
 
 func cosineSimilarityTester(x []float32, y []float32) float64 {
@@ -243,13 +244,15 @@ func TestSemanticSimilarityCuda(t *testing.T) {
 	}
 
 	// we now write the fine-tuned pipeline back to disk as an onnx model
-	if e := session.Save("./models/testTrain.onnx"); e != nil {
+	if e := session.Save("./models/testTrain"); e != nil {
 		t.Fatal(e)
 	}
-	if _, err := os.Stat("./models/testTrain.onnx"); err != nil {
+	if exists, existsErr := util.FileExists("./models/testTrain"); existsErr != nil {
 		t.Fatal(err)
+	} else if !exists {
+		t.Fatal("model file ./models/testTrain does not exist")
 	}
-	if err = os.Remove("./models/testTrain.onnx"); err != nil {
+	if err = util.DeleteFile("./models/testTrain"); err != nil {
 		t.Fatal(err)
 	}
 }