Update trace.py

trace.py

@@ -12,7 +12,6 @@ def trace_wandb(config,
                 prompt,
                 completion,
                 result,
-                callback,
                 err_msg,
                 start_time_ms,
                 end_time_ms):
@@ -36,7 +35,6 @@ def trace_wandb(config,
                   "config": str(config)
                  } if (str(err_msg) == "") else {},
         outputs = {"result": str(result),
-                   "callback": str(callback),
                    "completion": str(completion)
                   } if (str(err_msg) == "") else {},
         start_time_ms = start_time_ms,