mypy fix - .to should not be called if np.array

kieranfraser · kieranfraser · commit 84c9e2b28629 · 2023-06-28T16:46:18.000+01:00
Signed-off-by: Kieran Fraser &lt;Kieran.Fraser@ibm.com&gt;
diff --git a/art/estimators/object_detection/pytorch_detection_transformer.py b/art/estimators/object_detection/pytorch_detection_transformer.py
@@ -299,7 +299,7 @@ def _get_losses(
                 y_tensor = y  # type: ignore
 
             x_preprocessed, y_preprocessed = self._apply_preprocessing(
-                x.to(self.device), y=y_tensor, fit=False, no_grad=True
+                x, y=y_tensor, fit=False, no_grad=True
             )
 
             if self.clip_values is not None:

Original file line number	Diff line number	Diff line change
`@@ -299,7 +299,7 @@ def _get_losses(`
`299`	`299`	`y_tensor = y # type: ignore`
`300`	`300`
`301`	`301`	`x_preprocessed, y_preprocessed = self._apply_preprocessing(`
`302`		`- x.to(self.device), y=y_tensor, fit=False, no_grad=True`
	`302`	`+ x, y=y_tensor, fit=False, no_grad=True`
`303`	`303`	`)`
`304`	`304`
`305`	`305`	`if self.clip_values is not None:`