Skip to content

Commit 5a668f3

Browse files
Cast token count from size_t to int in C#
1 parent 6913f9f commit 5a668f3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/csharp/TestOnnxRuntimeGenAIAPI.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ public void TestGreedySearch()
163163
Assert.False(generator.IsDone());
164164
Assert.Equal(generatorParams.GetSearchNumber("max_length"), maxLength);
165165
Assert.Equal(generatorParams.GetSearchBool("early_stopping"), true);
166-
Assert.Equal(generator.TokenCount(), generator.GetSequence(0).Length);
166+
Assert.Equal((int)generator.TokenCount(), generator.GetSequence(0).Length);
167167

168168
while (!generator.IsDone())
169169
{
@@ -175,7 +175,7 @@ public void TestGreedySearch()
175175
var sequence = generator.GetSequence(i).ToArray();
176176
var expectedSequence = expectedOutput.Skip((int)i * (int)maxLength).Take((int)maxLength);
177177
Assert.Equal(expectedSequence, sequence);
178-
Assert.Equal(generator.TokenCount(), generator.GetSequence(i).Length);
178+
Assert.Equal((int)generator.TokenCount(), generator.GetSequence(i).Length);
179179
}
180180
}
181181
}

0 commit comments

Comments
 (0)