@@ -261,12 +261,12 @@ def apply_recursively(obj: Any) -> Any:
261
261
class RandomCrop (Transform ):
262
262
def __init__ (
263
263
self ,
264
- size = Union [int , Sequence [int ]],
265
- padding : Sequence [int ] = None ,
264
+ size : Union [int , Sequence [int ]],
265
+ padding : Sequence [int ],
266
266
pad_if_needed : bool = False ,
267
267
fill : Union [int , str , Sequence [int ]] = 0 ,
268
- padding_mode : Union [ str , Literal ["constant" , "edge" , "reflect" , "symmetric" ] ] = "constant" ,
269
- ):
268
+ padding_mode : Literal ["constant" , "edge" , "reflect" , "symmetric" ] = "constant" ,
269
+ ) -> None :
270
270
super ().__init__ ()
271
271
self .size = _setup_size (size , error_msg = "Please provide only two dimensions (h, w) for size." )
272
272
@@ -318,7 +318,7 @@ def forward(self, *inputs: Any) -> Any:
318
318
sample ,
319
319
output_size = self .size ,
320
320
image_size = get_image_dimensions (sample ),
321
- padding = self .padding ,
321
+ padding = cast ( List [ int ], tuple ( self .padding )) ,
322
322
pad_if_needed = self .pad_if_needed ,
323
323
fill = self .fill ,
324
324
padding_mode = self .padding_mode ,
@@ -329,7 +329,7 @@ def forward(self, *inputs: Any) -> Any:
329
329
sample ,
330
330
output_size = self .size ,
331
331
image_size = get_image_dimensions (sample ),
332
- padding = self .padding ,
332
+ padding = cast ( List [ int ], tuple ( self .padding )) ,
333
333
pad_if_needed = self .pad_if_needed ,
334
334
fill = self .fill ,
335
335
padding_mode = self .padding_mode ,
@@ -339,9 +339,9 @@ def forward(self, *inputs: Any) -> Any:
339
339
sample ,
340
340
output_size = self .size ,
341
341
image_size = get_image_dimensions (sample ),
342
- padding = self .padding ,
342
+ padding = cast ( List [ int ], tuple ( self .padding )) ,
343
343
pad_if_needed = self .pad_if_needed ,
344
- fill = self .fill ,
344
+ fill = self .fill , # TODO: should be converted to number
345
345
padding_mode = self .padding_mode ,
346
346
)
347
347
0 commit comments