Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 23, 2023

Commit

b552593

1 Parent(s): 08b6d98

Update trace.py

Files changed (1) hide show

trace.py CHANGED Viewed

@@ -4,7 +4,8 @@ from wandb.sdk.data_types.trace_tree import Trace
 WANDB_API_KEY = os.environ["WANDB_API_KEY"]
-def wandb_trace(is_rag_off,
                 prompt,
                 completion,
                 result,

 WANDB_API_KEY = os.environ["WANDB_API_KEY"]
+def wandb_trace(config,
+                is_rag_off,
                 prompt,
                 completion,
                 result,