Skip to content

Commit e2dc28d

Browse files
authored
Explicitly close encoded file (#755)
1 parent 7945e6a commit e2dc28d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/torchcodec/_core/Encoder.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,11 @@ AVSampleFormat findBestOutputSampleFormat(const AVCodec& avCodec) {
9898

9999
} // namespace
100100

101-
AudioEncoder::~AudioEncoder() {}
101+
AudioEncoder::~AudioEncoder() {
102+
if (avFormatContext_ && avFormatContext_->pb && !avioContextHolder_) {
103+
avio_close(avFormatContext_->pb);
104+
}
105+
}
102106

103107
AudioEncoder::AudioEncoder(
104108
const torch::Tensor& samples,

0 commit comments

Comments
 (0)