Skip to content

Commit 641bede

Browse files
authored
Rename gorse-benchmark tool to gorse-cli (#1163)
1 parent 0add752 commit 641bede

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed
Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,16 @@ import (
4545
)
4646

4747
var rootCmd = &cobra.Command{
48-
Use: "gorse-benchmark",
49-
Short: "Gorse Benchmarking Tool",
48+
Use: "gorse-cli",
49+
Short: "Gorse command line tool",
50+
Run: func(cmd *cobra.Command, args []string) {
51+
_ = cmd.Help()
52+
},
5053
}
5154

52-
var llmCmd = &cobra.Command{
53-
Use: "llm",
54-
Short: "Benchmark LLM models",
55+
var benchLLMCmd = &cobra.Command{
56+
Use: "bench-llm",
57+
Short: "Benchmark LLM models for ranking",
5558
Run: func(cmd *cobra.Command, args []string) {
5659
// Load configuration
5760
configPath, _ := cmd.Flags().GetString("config")
@@ -357,8 +360,8 @@ func EvaluateLLM(cfg *config.Config, train, test dataset.CFSplit, scores *sync.M
357360

358361
func init() {
359362
rootCmd.PersistentFlags().StringP("config", "c", "", "Path to configuration file")
360-
rootCmd.AddCommand(llmCmd)
361-
llmCmd.PersistentFlags().IntP("shots", "s", math.MaxInt, "Number of shots for each user")
363+
rootCmd.AddCommand(benchLLMCmd)
364+
benchLLMCmd.PersistentFlags().IntP("shots", "s", math.MaxInt, "Number of shots for each user")
362365
}
363366

364367
func main() {

0 commit comments

Comments
 (0)