Add comment

pytorch · May 31, 2024 · ff40253 · ff40253
1 parent c5810ff
commit ff40253
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/io/image.py b/torchvision/io/image.py
@@ -235,7 +235,7 @@ def write_jpeg(input: torch.Tensor, filename: str, quality: int = 75):
     if not torch.jit.is_scripting() and not torch.jit.is_tracing():
         _log_api_usage_once(write_jpeg)
     output = encode_jpeg(input, quality)
-    assert isinstance(output, torch.Tensor)
+    assert isinstance(output, torch.Tensor)  # Needed for torchscript
     write_file(filename, output)