fix

Author: NicolasHug

src/torchcodec/_core/Encoder.h

@@ -11,12 +11,6 @@ class AudioEncoder {
 
   AudioEncoder(
       const torch::Tensor& samples,
-      // TODO-ENCODING: update this comment when we support an output sample
-      // rate. This will become the input sample rate.
-      // The *output* sample rate. We can't really decide for the user what it
-      // should be. Particularly, the sample rate of the input samples should
-      // match this, and that's up to the user. If sample rates don't match,
-      // encoding will still work but audio will be distorted.
       int sampleRate,
       std::string_view fileName,
       const AudioStreamOptions& audioStreamOptions);

test/test_encoders.py

@@ -2,6 +2,7 @@
 import os
 import re
 import subprocess
+import sys
 from functools import partial
 from pathlib import Path
 
@@ -12,6 +13,7 @@
 from torchcodec.encoders import AudioEncoder
 
 from .utils import (
+    assert_tensor_close_on_at_least,
     get_ffmpeg_major_version,
     in_fbcode,
     NASA_AUDIO_MP3,
@@ -266,9 +268,7 @@ def test_against_cli(
         if sample_rate != asset.sample_rate:
             rtol, atol = 0, 1e-3
             if sys.platform == "darwin":
-                assert_close = partial(
-                    utils.assert_tensor_close_on_at_least, percentage=99
-                )
+                assert_close = partial(assert_tensor_close_on_at_least, percentage=99)
         elif format == "wav":
             rtol, atol = 0, 1e-4
         elif format == "mp3" and asset is SINE_MONO_S32 and num_channels == 2: