nits

Author: Dan-Flores

src/torchcodec/_core/Encoder.cpp

@@ -917,17 +917,6 @@ UniqueAVFrame VideoEncoder::convertTensorToAVFrame(
       "Expected 3D RGB tensor (CHW format), got shape: ",
       frame.sizes());
 
-  // These are all already set in initializeEncoder?
-  // inHeight_ = static_cast<int>(tensor.sizes()[1]);
-  // inWidth_ = static_cast<int>(tensor.sizes()[2]);
-
-  // // For now, reuse input dimensions as output dimensions
-  // outWidth_ = inWidth_;
-  // outHeight_ = inHeight_;
-
-  // // Input format is RGB planar (AV_PIX_FMT_GBRP after channel reordering)
-  // inPixelFormat_ = AV_PIX_FMT_GBRP;
-
   // Initialize and cache scaling context if it does not exist
   if (!swsContext_) {
     swsContext_.reset(sws_getContext(

src/torchcodec/_core/custom_ops.cpp

@@ -415,6 +415,7 @@ void _add_video_stream(
   }
 
   validateDeviceInterface(std::string(device), std::string(device_variant));
+
   videoStreamOptions.device = torch::Device(std::string(device));
   videoStreamOptions.deviceVariant = device_variant;
 

src/torchcodec/encoders/_video_encoder.py

@@ -19,12 +19,7 @@ class VideoEncoder:
         frame_rate (float): The frame rate of the **input** ``frames``. Also defines the encoded **output** frame rate.
     """
 
-    def __init__(
-        self,
-        frames: Tensor,
-        *,
-        frame_rate: float,
-    ):
+    def __init__(self, frames: Tensor, *, frame_rate: float):
         torch._C._log_api_usage_once("torchcodec.encoders.VideoEncoder")
         if not isinstance(frames, Tensor):
             raise ValueError(f"Expected frames to be a Tensor, got {type(frames) = }.")

test/test_encoders.py

@@ -10,6 +10,7 @@
 import pytest
 import torch
 from torchcodec.decoders import AudioDecoder, VideoDecoder
+
 from torchcodec.encoders import AudioEncoder, VideoEncoder
 
 from .utils import (