fix: update generate save path

terryyz · terryyz · commit 4a23976c8d06 · 2024-07-06T21:18:30.000+08:00
diff --git a/bigcodebench/generate.py b/bigcodebench/generate.py
@@ -145,7 +145,7 @@ def main():
         trust_remote_code=args.trust_remote_code
     )
     
-    extra = "-" + args.subset if args.subset else ""
+    extra = "-" + args.subset if args.subset != "full" else ""
     if not args.save_path:
         save_path = args.model.replace("/", "--") + f"--bigcodebench{extra}-{args.split}--{args.backend}-{args.temperature}-{args.n_samples}.jsonl"
     else:

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,7 @@ def main():`
`145`	`145`	`trust_remote_code=args.trust_remote_code`
`146`	`146`	`)`
`147`	`147`
`148`		`- extra = "-" + args.subset if args.subset else ""`
	`148`	`+ extra = "-" + args.subset if args.subset != "full" else ""`
`149`	`149`	`if not args.save_path:`
`150`	`150`	`save_path = args.model.replace("/", "--") + f"--bigcodebench{extra}-{args.split}--{args.backend}-{args.temperature}-{args.n_samples}.jsonl"`
`151`	`151`	`else:`