From c0fa2236cfa539c7fcb79a20e0770933ea7d9a22 Mon Sep 17 00:00:00 2001 From: Roy Han Date: Wed, 3 Jul 2024 12:47:57 -0700 Subject: [PATCH] integration float32 --- integration/embed_test.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/integration/embed_test.go b/integration/embed_test.go index 01baa265c..f13fd388a 100644 --- a/integration/embed_test.go +++ b/integration/embed_test.go @@ -33,8 +33,8 @@ func TestAllMiniLMEmbed(t *testing.T) { t.Fatalf("expected 384 floats, got %d", len(res.Embeddings[0])) } - if res.Embeddings[0][0] != 0.010071029038540258 { - t.Fatalf("expected 0.010071029038540258, got %f", res.Embeddings[0][0]) + if res.Embeddings[0][0] != 0.010071029 { + t.Fatalf("expected 0.010071029, got %f", res.Embeddings[0][0]) } } @@ -61,8 +61,8 @@ func TestAllMiniLMBatchEmbed(t *testing.T) { t.Fatalf("expected 384 floats, got %d", len(res.Embeddings[0])) } - if res.Embeddings[0][0] != 0.010071029038540258 || res.Embeddings[1][0] != -0.00980270794235093 { - t.Fatalf("expected 0.010071029038540258 and -0.00980270794235093, got %f and %f", res.Embeddings[0][0], res.Embeddings[1][0]) + if res.Embeddings[0][0] != 0.010071029 || res.Embeddings[1][0] != -0.0098027075 { + t.Fatalf("expected 0.010071029 and -0.0098027075, got %f and %f", res.Embeddings[0][0], res.Embeddings[1][0]) } }