16 lines
606 B
Lua
16 lines
606 B
Lua
function add_labels(tag, ts, record)
|
|
record["job_id"] = os.getenv("FB_JOB_ID") or record["job_id"] or "unknown"
|
|
record["user"] = os.getenv("FB_USER") or record["user"] or "unknown"
|
|
record["model"] = os.getenv("FB_MODEL") or record["model"] or "unknown"
|
|
record["gpu_id"] = os.getenv("FB_GPU_ID") or record["gpu_id"] or "na"
|
|
local p = record["log_path"] or ""
|
|
if string.find(p, "/logs/infer/") then
|
|
record["role"] = "infer"
|
|
elseif string.find(p, "/logs/train/") then
|
|
record["role"] = "train"
|
|
else
|
|
record["role"] = record["role"] or "app"
|
|
end
|
|
return 1, ts, record
|
|
end
|