how to add length of audiofile to diarization result?

[freshpearYoon]

Hi, I want to add lenght of audio file to rttm file when dumping diarization output to rttm format.
Here's the code.
"
diarization = pipeline("../segments/%s"%(file))
# dump the diarization output to disk using RTTM format
with open("../rttms/"+name+".rttm", "w") as rttm:
diarization.write_rttm(rttm)
"
Is there any function or feature that I can add length of audio to diarization output?