Fix stopPts logic

Author: NicolasHug

src/torchcodec/decoders/_core/VideoDecoder.cpp

@@ -890,16 +890,15 @@ VideoDecoder::AudioFramesOutput VideoDecoder::getFramesPlayedInRangeAudio(
     std::optional<double> stopSecondsOptional) {
   validateActiveStream(AVMEDIA_TYPE_AUDIO);
 
-  double stopSeconds =
-      stopSecondsOptional.value_or(std::numeric_limits<double>::max());
-
-  TORCH_CHECK(
-      startSeconds <= stopSeconds,
-      "Start seconds (" + std::to_string(startSeconds) +
-          ") must be less than or equal to stop seconds (" +
-          std::to_string(stopSeconds) + ").");
+  if (stopSecondsOptional.has_value()) {
+    TORCH_CHECK(
+        startSeconds <= *stopSecondsOptional,
+        "Start seconds (" + std::to_string(startSeconds) +
+            ") must be less than or equal to stop seconds (" +
+            std::to_string(*stopSecondsOptional) + ").");
+  }
 
-  if (startSeconds == stopSeconds) {
+  if (stopSecondsOptional.has_value() && startSeconds == *stopSecondsOptional) {
     // For consistency with video
     return AudioFramesOutput{torch::empty({0, 0}), 0.0};
   }
@@ -921,7 +920,9 @@ VideoDecoder::AudioFramesOutput VideoDecoder::getFramesPlayedInRangeAudio(
   std::vector<torch::Tensor> frames;
 
   std::optional<double> firstFramePtsSeconds = std::nullopt;
-  auto stopPts = secondsToClosestPts(stopSeconds, streamInfo.timeBase);
+  auto stopPts = stopSecondsOptional.has_value()
+      ? secondsToClosestPts(*stopSecondsOptional, streamInfo.timeBase)
+      : INT64_MAX;
   auto finished = false;
   while (!finished) {
     try {