Sfoglia il codice sorgente

last lint (hopefully)

Josh Yan 8 mesi fa
parent
commit
52ef79bb7d
3 ha cambiato i file con 5 aggiunte e 5 eliminazioni
  1. 2 2
      server/model_test.go
  2. 1 1
      server/routes_create_test.go
  3. 2 2
      server/sched_test.go

+ 2 - 2
server/model_test.go

@@ -145,7 +145,7 @@ func TestParseFromFileFromLayer(t *testing.T) {
 		t.Fatalf("failed to open file: %v", err)
 		t.Fatalf("failed to open file: %v", err)
 	}
 	}
 	defer file.Close()
 	defer file.Close()
-	if err := llm.WriteGGUF(file, llm.KV{"general.architecture": "gemma"}, []llm.Tensor{}, func(api.ProgressResponse){}); err != nil {
+	if err := llm.WriteGGUF(file, llm.KV{"general.architecture": "gemma"}, []llm.Tensor{}, func(api.ProgressResponse) {}); err != nil {
 		t.Fatalf("failed to write gguf: %v", err)
 		t.Fatalf("failed to write gguf: %v", err)
 	}
 	}
 
 
@@ -197,7 +197,7 @@ func TestParseLayerFromCopy(t *testing.T) {
 	defer file2.Close()
 	defer file2.Close()
 
 
 	for range 5 {
 	for range 5 {
-		if err := llm.WriteGGUF(file2, llm.KV{"general.architecture": "gemma"}, []llm.Tensor{}, func(api.ProgressResponse){}); err != nil {
+		if err := llm.WriteGGUF(file2, llm.KV{"general.architecture": "gemma"}, []llm.Tensor{}, func(api.ProgressResponse) {}); err != nil {
 			t.Fatalf("failed to write gguf: %v", err)
 			t.Fatalf("failed to write gguf: %v", err)
 		}
 		}
 	}
 	}

+ 1 - 1
server/routes_create_test.go

@@ -30,7 +30,7 @@ func createBinFile(t *testing.T, kv map[string]any, ti []llm.Tensor) string {
 	}
 	}
 	defer f.Close()
 	defer f.Close()
 
 
-	if err := llm.WriteGGUF(f, kv, ti, func(api.ProgressResponse){}); err != nil {
+	if err := llm.WriteGGUF(f, kv, ti, func(api.ProgressResponse) {}); err != nil {
 		t.Fatal(err)
 		t.Fatal(err)
 	}
 	}
 
 

+ 2 - 2
server/sched_test.go

@@ -128,8 +128,8 @@ func newScenarioRequest(t *testing.T, ctx context.Context, modelName string, est
 	}, []llm.Tensor{
 	}, []llm.Tensor{
 		{Name: "blk.0.attn.weight", Kind: uint32(0), Offset: uint64(0), Shape: []uint64{1, 1, 1, 1}, WriterTo: bytes.NewReader(make([]byte, 32))},
 		{Name: "blk.0.attn.weight", Kind: uint32(0), Offset: uint64(0), Shape: []uint64{1, 1, 1, 1}, WriterTo: bytes.NewReader(make([]byte, 32))},
 		{Name: "output.weight", Kind: uint32(0), Offset: uint64(0), Shape: []uint64{1, 1, 1, 1}, WriterTo: bytes.NewReader(make([]byte, 32))},
 		{Name: "output.weight", Kind: uint32(0), Offset: uint64(0), Shape: []uint64{1, 1, 1, 1}, WriterTo: bytes.NewReader(make([]byte, 32))},
-	}, 
-	func(api.ProgressResponse){}))
+	},
+		func(api.ProgressResponse) {}))
 	require.NoError(t, err)
 	require.NoError(t, err)
 
 
 	fname := f.Name()
 	fname := f.Name()