Spaces:
Build error
Build error
Update trace.py
Browse files
trace.py
CHANGED
|
@@ -6,7 +6,6 @@ WANDB_API_KEY = os.environ["WANDB_API_KEY"]
|
|
| 6 |
|
| 7 |
def trace_wandb(config,
|
| 8 |
rag_option,
|
| 9 |
-
prompt_template,
|
| 10 |
prompt,
|
| 11 |
completion,
|
| 12 |
result,
|
|
@@ -22,7 +21,7 @@ def trace_wandb(config,
|
|
| 22 |
status_code = "success" if (str(err_msg) == "") else "error",
|
| 23 |
status_message = str(err_msg),
|
| 24 |
inputs = {"prompt": prompt,
|
| 25 |
-
"prompt_template": prompt_template,
|
| 26 |
"rag_option": rag_option,
|
| 27 |
"config": str(config)
|
| 28 |
} if (str(err_msg) == "") else {},
|
|
|
|
| 6 |
|
| 7 |
def trace_wandb(config,
|
| 8 |
rag_option,
|
|
|
|
| 9 |
prompt,
|
| 10 |
completion,
|
| 11 |
result,
|
|
|
|
| 21 |
status_code = "success" if (str(err_msg) == "") else "error",
|
| 22 |
status_message = str(err_msg),
|
| 23 |
inputs = {"prompt": prompt,
|
| 24 |
+
#"prompt_template": prompt_template,
|
| 25 |
"rag_option": rag_option,
|
| 26 |
"config": str(config)
|
| 27 |
} if (str(err_msg) == "") else {},
|