restore int idiom

datumbox · datumbox · commit 537e650f6373 · 2022-10-26T16:41:43.000+01:00
diff --git a/torchvision/prototype/transforms/_geometry.py b/torchvision/prototype/transforms/_geometry.py
@@ -448,12 +448,12 @@ def _get_params(self, flat_inputs: List[Any]) -> Dict[str, Any]:
         needs_pad = any(padding)
 
         needs_vert_crop, top = (
-            (True, torch.randint(0, padded_height - cropped_height + 1, size=()).item())
+            (True, int(torch.randint(0, padded_height - cropped_height + 1, size=())))
             if padded_height > cropped_height
             else (False, 0)
         )
         needs_horz_crop, left = (
-            (True, torch.randint(0, padded_width - cropped_width + 1, size=()).item())
+            (True, int(torch.randint(0, padded_width - cropped_width + 1, size=()).item()))
             if padded_width > cropped_width
             else (False, 0)
         )

Original file line number	Diff line number	Diff line change
`@@ -448,12 +448,12 @@ def _get_params(self, flat_inputs: List[Any]) -> Dict[str, Any]:`
`448`	`448`	`needs_pad = any(padding)`
`449`	`449`
`450`	`450`	`needs_vert_crop, top = (`
`451`		`- (True, torch.randint(0, padded_height - cropped_height + 1, size=()).item())`
	`451`	`+ (True, int(torch.randint(0, padded_height - cropped_height + 1, size=())))`
`452`	`452`	`if padded_height > cropped_height`
`453`	`453`	`else (False, 0)`
`454`	`454`	`)`
`455`	`455`	`needs_horz_crop, left = (`
`456`		`- (True, torch.randint(0, padded_width - cropped_width + 1, size=()).item())`
	`456`	`+ (True, int(torch.randint(0, padded_width - cropped_width + 1, size=()).item()))`
`457`	`457`	`if padded_width > cropped_width`
`458`	`458`	`else (False, 0)`
`459`	`459`	`)`