@@ -207,15 +207,15 @@ func TestGetContextLength(t *testing.T) {
207207 backendCfg : & inference.BackendConfiguration {
208208 ContextSize : int32ptr (4096 ),
209209 },
210- expectedValue : ptrUint64 (4096 ),
210+ expectedValue : float64ptr (4096 ),
211211 },
212212 {
213213 name : "model config only" ,
214214 modelCfg : types.Config {
215215 ContextSize : int32ptr (8192 ),
216216 },
217217 backendCfg : nil ,
218- expectedValue : ptrUint64 (8192 ),
218+ expectedValue : float64ptr (8192 ),
219219 },
220220 {
221221 name : "model config takes precedence" ,
@@ -225,7 +225,7 @@ func TestGetContextLength(t *testing.T) {
225225 backendCfg : & inference.BackendConfiguration {
226226 ContextSize : int32ptr (4096 ),
227227 },
228- expectedValue : ptrUint64 (16384 ),
228+ expectedValue : float64ptr (16384 ),
229229 },
230230 {
231231 name : "zero context size in backend config returns nil" ,
@@ -249,7 +249,7 @@ func TestGetContextLength(t *testing.T) {
249249 }
250250}
251251
252- func ptrUint64 (v uint64 ) * uint64 {
252+ func float64ptr (v uint64 ) * uint64 {
253253 return & v
254254}
255255
0 commit comments