Skip to content

Commit 7b46289

Browse files
committed
Fix mypy alerts
Signed-off-by: Beat Buesser <[email protected]>
1 parent f883f7c commit 7b46289

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

art/estimators/object_detection/pytorch_object_detector.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -245,16 +245,15 @@ def _preprocess_and_convert_inputs(
245245
x_preprocessed, y_preprocessed = self._apply_preprocessing(x=x, y=y, fit=fit, no_grad=no_grad)
246246

247247
# Convert inputs into tensor
248-
x_preprocessed_tensor_2: torch.Tensor
249-
x_preprocessed_tensor_2, y_preprocessed_tensor = cast_inputs_to_pt(x_preprocessed, y_preprocessed)
248+
x_preprocessed_tensor, y_preprocessed_tensor = cast_inputs_to_pt(x_preprocessed, y_preprocessed)
250249

251250
if not self.channels_first:
252-
x_preprocessed_tensor_2 = torch.permute(x_preprocessed_tensor_2, (0, 3, 1, 2))
253-
x_preprocessed_tensor_2 = x_preprocessed_tensor_2 / torch.tensor(norm_factor, device=self.device)
251+
x_preprocessed_tensor = torch.permute(x_preprocessed_tensor, (0, 3, 1, 2))
252+
x_preprocessed_tensor = x_preprocessed_tensor / torch.tensor(norm_factor, device=self.device)
254253

255254
# Set gradients
256255
if not no_grad:
257-
x_preprocessed_tensor_2.requires_grad = True
256+
x_preprocessed_tensor.requires_grad = True
258257

259258
else:
260259
raise NotImplementedError("Combination of inputs and preprocessing not supported.")

0 commit comments

Comments
 (0)