Change 'torch.Tensor's to 'torch.tensor' (#2601)

sadra-barikbin · web-flow · commit 9175f392a138 · 2022-06-13T00:46:00.000+02:00
* Change 'torch.Tensor's to 'torch.tensor' Closes #2600 * Fix type-related errors in doctests * Fix visdom_logger Mypy issue (cherry picked from commit 8e8fad14c9e3e221ccf5b62d797603b096ca01ca) * Fix test_visdom_logger and do a refactor in it
diff --git a/ignite/contrib/handlers/visdom_logger.py b/ignite/contrib/handlers/visdom_logger.py
@@ -528,6 +528,9 @@ def __call__(self, engine: Engine, logger: VisdomLogger, event_name: Union[str,
         global_step = engine.state.get_event_attrib_value(event_name)
         tag_prefix = f"{self.tag}/" if self.tag else ""
         for name, p in self.model.named_parameters():
+            if p.grad is None:
+                continue
+
             name = name.replace(".", "/")
             k = f"{tag_prefix}grads_{self.reduction.__name__}/{name}"
             v = self.reduction(p.grad)
diff --git a/ignite/contrib/metrics/average_precision.py b/ignite/contrib/metrics/average_precision.py
@@ -48,7 +48,7 @@ def activated_output_transform(output):
 
         .. testcode::
 
-            y_pred = torch.Tensor([[0.79, 0.21], [0.30, 0.70], [0.46, 0.54], [0.16, 0.84]])
+            y_pred = torch.tensor([[0.79, 0.21], [0.30, 0.70], [0.46, 0.54], [0.16, 0.84]])
             y_true = torch.tensor([[1, 1], [1, 1], [0, 1], [0, 1]])
 
             avg_precision = AveragePrecision()
diff --git a/ignite/contrib/metrics/cohen_kappa.py b/ignite/contrib/metrics/cohen_kappa.py
@@ -37,8 +37,8 @@ class CohenKappa(EpochMetric):
 
             metric = CohenKappa()
             metric.attach(default_evaluator, 'ck')
-            y_true = torch.Tensor([2, 0, 2, 2, 0, 1])
-            y_pred = torch.Tensor([0, 0, 2, 2, 0, 2])
+            y_true = torch.tensor([2, 0, 2, 2, 0, 1])
+            y_pred = torch.tensor([0, 0, 2, 2, 0, 2])
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['ck'])
 
diff --git a/ignite/contrib/metrics/regression/canberra_metric.py b/ignite/contrib/metrics/regression/canberra_metric.py
@@ -49,7 +49,7 @@ class CanberraMetric(_BaseRegression):
 
                 metric = CanberraMetric()
                 metric.attach(default_evaluator, 'canberra')
-                y_pred = torch.Tensor([[3.8], [9.9], [-5.4], [2.1]])
+                y_pred = torch.tensor([[3.8], [9.9], [-5.4], [2.1]])
                 y_true = y_pred * 1.5
                 state = default_evaluator.run([[y_pred, y_true]])
                 print(state.metrics['canberra'])
diff --git a/ignite/contrib/metrics/regression/fractional_absolute_error.py b/ignite/contrib/metrics/regression/fractional_absolute_error.py
@@ -46,7 +46,7 @@ class FractionalAbsoluteError(_BaseRegression):
 
                 metric = FractionalAbsoluteError()
                 metric.attach(default_evaluator, 'fractional_abs_error')
-                y_pred = torch.Tensor([[3.8], [9.9], [-5.4], [2.1]])
+                y_pred = torch.tensor([[3.8], [9.9], [-5.4], [2.1]])
                 y_true = y_pred * 0.8
                 state = default_evaluator.run([[y_pred, y_true]])
                 print(state.metrics['fractional_abs_error'])
diff --git a/ignite/contrib/metrics/regression/fractional_bias.py b/ignite/contrib/metrics/regression/fractional_bias.py
@@ -46,7 +46,7 @@ class FractionalBias(_BaseRegression):
 
                 metric = FractionalBias()
                 metric.attach(default_evaluator, 'fractional_bias')
-                y_pred = torch.Tensor([[3.8], [9.9], [5.4], [2.1]])
+                y_pred = torch.tensor([[3.8], [9.9], [5.4], [2.1]])
                 y_true = y_pred * 1.5
                 state = default_evaluator.run([[y_pred, y_true]])
                 print(state.metrics['fractional_bias'])
diff --git a/ignite/contrib/metrics/regression/geometric_mean_absolute_error.py b/ignite/contrib/metrics/regression/geometric_mean_absolute_error.py
@@ -46,7 +46,7 @@ class GeometricMeanAbsoluteError(_BaseRegression):
 
                 metric = GeometricMeanAbsoluteError()
                 metric.attach(default_evaluator, 'gmae')
-                y_pred = torch.Tensor([[3.8], [9.9], [-5.4], [2.1]])
+                y_pred = torch.tensor([[3.8], [9.9], [-5.4], [2.1]])
                 y_true = y_pred * 1.5
                 state = default_evaluator.run([[y_pred, y_true]])
                 print(state.metrics['gmae'])
diff --git a/ignite/contrib/metrics/regression/geometric_mean_relative_absolute_error.py b/ignite/contrib/metrics/regression/geometric_mean_relative_absolute_error.py
@@ -60,7 +60,7 @@ class GeometricMeanRelativeAbsoluteError(_BaseRegression):
 
             metric = GeometricMeanRelativeAbsoluteError()
             metric.attach(default_evaluator, 'gmare')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['gmare'])
diff --git a/ignite/contrib/metrics/regression/manhattan_distance.py b/ignite/contrib/metrics/regression/manhattan_distance.py
@@ -45,7 +45,7 @@ class ManhattanDistance(_BaseRegression):
 
             metric = ManhattanDistance()
             metric.attach(default_evaluator, 'manhattan')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['manhattan'])
diff --git a/ignite/contrib/metrics/regression/maximum_absolute_error.py b/ignite/contrib/metrics/regression/maximum_absolute_error.py
@@ -46,7 +46,7 @@ class MaximumAbsoluteError(_BaseRegression):
 
             metric = MaximumAbsoluteError()
             metric.attach(default_evaluator, 'mae')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['mae'])
diff --git a/ignite/contrib/metrics/regression/mean_absolute_relative_error.py b/ignite/contrib/metrics/regression/mean_absolute_relative_error.py
@@ -46,7 +46,7 @@ class MeanAbsoluteRelativeError(_BaseRegression):
 
             metric = MeanAbsoluteRelativeError()
             metric.attach(default_evaluator, 'mare')
-            y_true = torch.Tensor([1, 2, 3, 4, 5])
+            y_true = torch.tensor([1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['mare'])
diff --git a/ignite/contrib/metrics/regression/mean_error.py b/ignite/contrib/metrics/regression/mean_error.py
@@ -46,7 +46,7 @@ class MeanError(_BaseRegression):
 
             metric = MeanError()
             metric.attach(default_evaluator, 'me')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['me'])
diff --git a/ignite/contrib/metrics/regression/mean_normalized_bias.py b/ignite/contrib/metrics/regression/mean_normalized_bias.py
@@ -46,7 +46,7 @@ class MeanNormalizedBias(_BaseRegression):
 
             metric = MeanNormalizedBias()
             metric.attach(default_evaluator, 'mnb')
-            y_true = torch.Tensor([1, 2, 3, 4, 5])
+            y_true = torch.tensor([1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['mnb'])
diff --git a/ignite/contrib/metrics/regression/median_absolute_error.py b/ignite/contrib/metrics/regression/median_absolute_error.py
@@ -52,7 +52,7 @@ class MedianAbsoluteError(EpochMetric):
 
             metric = MedianAbsoluteError()
             metric.attach(default_evaluator, 'mae')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0, 1, 2, 3, 4, 5])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['mae'])
diff --git a/ignite/contrib/metrics/regression/median_absolute_percentage_error.py b/ignite/contrib/metrics/regression/median_absolute_percentage_error.py
@@ -50,7 +50,7 @@ class MedianAbsolutePercentageError(EpochMetric):
 
             metric = MedianAbsolutePercentageError()
             metric.attach(default_evaluator, 'mape')
-            y_true = torch.Tensor([1, 2, 3, 4, 5])
+            y_true = torch.tensor([1, 2, 3, 4, 5])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['mape'])
diff --git a/ignite/contrib/metrics/regression/median_relative_absolute_error.py b/ignite/contrib/metrics/regression/median_relative_absolute_error.py
@@ -50,7 +50,7 @@ class MedianRelativeAbsoluteError(EpochMetric):
 
             metric = MedianRelativeAbsoluteError()
             metric.attach(default_evaluator, 'mrae')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['mrae'])
diff --git a/ignite/contrib/metrics/regression/r2_score.py b/ignite/contrib/metrics/regression/r2_score.py
@@ -44,7 +44,7 @@ class R2Score(_BaseRegression):
 
             metric = R2Score()
             metric.attach(default_evaluator, 'r2')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['r2'])
diff --git a/ignite/contrib/metrics/regression/wave_hedges_distance.py b/ignite/contrib/metrics/regression/wave_hedges_distance.py
@@ -45,7 +45,7 @@ class WaveHedgesDistance(_BaseRegression):
 
             metric = WaveHedgesDistance()
             metric.attach(default_evaluator, 'whd')
-            y_true = torch.Tensor([0, 1, 2, 3, 4, 5])
+            y_true = torch.tensor([0., 1., 2., 3., 4., 5.])
             y_pred = y_true * 0.75
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics['whd'])
diff --git a/ignite/metrics/accumulation.py b/ignite/metrics/accumulation.py
@@ -225,7 +225,7 @@ class GeometricAverage(VariableAccumulation):
             metric = GeometricAverage()
             metric.attach(default_evaluator, 'avg')
             # Case 1. input is er
-            data = torch.Tensor([1, 2, 3])
+            data = torch.tensor([1, 2, 3])
             state = default_evaluator.run(data)
             print(state.metrics['avg'])
 
diff --git a/ignite/metrics/mean_pairwise_distance.py b/ignite/metrics/mean_pairwise_distance.py
@@ -44,7 +44,7 @@ class MeanPairwiseDistance(Metric):
 
             metric = MeanPairwiseDistance(p=4)
             metric.attach(default_evaluator, 'mpd')
-            preds = torch.Tensor([
+            preds = torch.tensor([
                 [1, 2, 4, 1],
                 [2, 3, 1, 5],
                 [1, 3, 5, 1],
diff --git a/ignite/metrics/precision.py b/ignite/metrics/precision.py
@@ -286,7 +286,7 @@ class Precision(_BasePrecisionRecall):
             weighted_metric.attach(default_evaluator, "weighted precision")
 
             y_true = torch.tensor([2, 0, 2, 1, 0])
-            y_pred = torch.Tensor([
+            y_pred = torch.tensor([
                 [0.0266, 0.1719, 0.3055],
                 [0.6886, 0.3978, 0.8176],
                 [0.9230, 0.0197, 0.8395],
@@ -320,14 +320,14 @@ class Precision(_BasePrecisionRecall):
             weighted_metric.attach(default_evaluator, "weighted precision")
             samples_metric.attach(default_evaluator, "samples precision")
 
-            y_true = torch.Tensor([
+            y_true = torch.tensor([
                 [0, 0, 1],
                 [0, 0, 0],
                 [0, 0, 0],
                 [1, 0, 0],
                 [0, 1, 1],
             ])
-            y_pred = torch.Tensor([
+            y_pred = torch.tensor([
                 [1, 1, 0],
                 [1, 0, 1],
                 [1, 0, 0],
@@ -361,7 +361,7 @@ def thresholded_output_transform(output):
             metric = Precision(output_transform=thresholded_output_transform)
             metric.attach(default_evaluator, "precision")
             y_true = torch.tensor([1, 0, 1, 1, 0, 1])
-            y_pred = torch.Tensor([0.6, 0.2, 0.9, 0.4, 0.7, 0.65])
+            y_pred = torch.tensor([0.6, 0.2, 0.9, 0.4, 0.7, 0.65])
             state = default_evaluator.run([[y_pred, y_true]])
             print(state.metrics["precision"])
 
diff --git a/tests/ignite/contrib/handlers/test_clearml_logger.py b/tests/ignite/contrib/handlers/test_clearml_logger.py
@@ -49,7 +49,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 def test_optimizer_params():
 
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=ClearMLLogger)
     mock_logger.clearml_logger = MagicMock()
diff --git a/tests/ignite/contrib/handlers/test_mlflow_logger.py b/tests/ignite/contrib/handlers/test_mlflow_logger.py
@@ -64,7 +64,7 @@ def test_output_handler_metric_names():
     wrapper = OutputHandler("tag", metric_names=["a"])
 
     mock_engine = MagicMock()
-    mock_engine.state = State(metrics={"a": torch.Tensor([0.0, 1.0, 2.0, 3.0])})
+    mock_engine.state = State(metrics={"a": torch.tensor([0.0, 1.0, 2.0, 3.0])})
     mock_engine.state.iteration = 5
 
     mock_logger = MagicMock(spec=MLflowLogger)
@@ -216,7 +216,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 def test_optimizer_params():
 
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=MLflowLogger)
     mock_logger.log_metrics = MagicMock()
diff --git a/tests/ignite/contrib/handlers/test_neptune_logger.py b/tests/ignite/contrib/handlers/test_neptune_logger.py
@@ -34,7 +34,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 
 def test_optimizer_params():
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=NeptuneLogger)
     mock_logger.log_metric = MagicMock()
@@ -99,7 +99,7 @@ def test_output_handler_metric_names():
 
     mock_engine = MagicMock()
     mock_logger.log_metric = MagicMock()
-    mock_engine.state = State(metrics={"a": torch.Tensor([0.0, 1.0, 2.0, 3.0])})
+    mock_engine.state = State(metrics={"a": torch.tensor([0.0, 1.0, 2.0, 3.0])})
     mock_engine.state.iteration = 5
 
     mock_logger = MagicMock(spec=NeptuneLogger)
diff --git a/tests/ignite/contrib/handlers/test_polyaxon_logger.py b/tests/ignite/contrib/handlers/test_polyaxon_logger.py
@@ -66,7 +66,7 @@ def test_output_handler_metric_names():
     wrapper = OutputHandler("tag", metric_names=["a"])
 
     mock_engine = MagicMock()
-    mock_engine.state = State(metrics={"a": torch.Tensor([0.0, 1.0, 2.0, 3.0])})
+    mock_engine.state = State(metrics={"a": torch.tensor([0.0, 1.0, 2.0, 3.0])})
     mock_engine.state.iteration = 5
 
     mock_logger = MagicMock(spec=PolyaxonLogger)
@@ -232,7 +232,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 def test_optimizer_params():
 
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=PolyaxonLogger)
     mock_logger.log_metrics = MagicMock()
diff --git a/tests/ignite/contrib/handlers/test_tensorboard_logger.py b/tests/ignite/contrib/handlers/test_tensorboard_logger.py
@@ -34,7 +34,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 def test_optimizer_params():
 
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=TensorboardLogger)
     mock_logger.writer = MagicMock()
@@ -104,7 +104,7 @@ def test_output_handler_metric_names():
     wrapper = OutputHandler("tag", metric_names=["a"])
 
     mock_engine = MagicMock()
-    mock_engine.state = State(metrics={"a": torch.Tensor([0.0, 1.0, 2.0, 3.0])})
+    mock_engine.state = State(metrics={"a": torch.tensor([0.0, 1.0, 2.0, 3.0])})
     mock_engine.state.iteration = 5
 
     mock_logger = MagicMock(spec=TensorboardLogger)
diff --git a/tests/ignite/contrib/handlers/test_visdom_logger.py b/tests/ignite/contrib/handlers/test_visdom_logger.py
@@ -32,7 +32,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 def test_optimizer_params():
 
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=VisdomLogger)
     mock_logger.vis = MagicMock()
@@ -181,7 +181,7 @@ def test_output_handler_metric_names(dirname):
     wrapper = OutputHandler("tag", metric_names=["a"])
 
     mock_engine = MagicMock()
-    mock_engine.state = State(metrics={"a": torch.Tensor([0.0, 1.0, 2.0, 3.0])})
+    mock_engine.state = State(metrics={"a": torch.tensor([0.0, 1.0, 2.0, 3.0])})
     mock_engine.state.iteration = 5
 
     mock_logger = MagicMock(spec=VisdomLogger)
@@ -785,25 +785,12 @@ def test_grads_scalar_handler_wrong_setup():
         wrapper(mock_engine, mock_logger, Events.ITERATION_STARTED)
 
 
-def test_grads_scalar_handler():
-    class DummyModel(torch.nn.Module):
-        def __init__(self):
-            super(DummyModel, self).__init__()
-            self.fc1 = torch.nn.Linear(10, 10)
-            self.fc2 = torch.nn.Linear(12, 12)
-            self.fc1.weight.data.zero_()
-            self.fc1.bias.data.zero_()
-            self.fc2.weight.data.fill_(1.0)
-            self.fc2.bias.data.fill_(1.0)
-
-    model = DummyModel()
-
-    def norm(x):
-        return 0.0
+def test_grads_scalar_handler(dummy_model_factory, norm_mock):
+    model = dummy_model_factory(with_grads=True, with_frozen_layer=False)
 
     # define test wrapper to test with and without optional tag
     def _test(tag=None):
-        wrapper = GradsScalarHandler(model, reduction=norm, tag=tag)
+        wrapper = GradsScalarHandler(model, reduction=norm_mock, tag=tag)
         mock_logger = MagicMock(spec=VisdomLogger)
         mock_logger.vis = MagicMock()
         mock_logger.executor = _DummyExecutor()
diff --git a/tests/ignite/contrib/handlers/test_wandb_logger.py b/tests/ignite/contrib/handlers/test_wandb_logger.py
@@ -26,7 +26,7 @@ def test_optimizer_params_handler_wrong_setup():
 
 
 def test_optimizer_params():
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=WandBLogger)
     mock_logger.log = MagicMock()
@@ -272,7 +272,7 @@ def test_output_handler_state_attrs():
 
 
 def test_wandb_close():
-    optimizer = torch.optim.SGD([torch.Tensor(0)], lr=0.01)
+    optimizer = torch.optim.SGD([torch.tensor(0.0)], lr=0.01)
     wrapper = OptimizerParamsHandler(optimizer=optimizer, param_name="lr")
     mock_logger = MagicMock(spec=WandBLogger)
     mock_logger.log = MagicMock()
diff --git a/tests/ignite/metrics/gan/test_fid.py b/tests/ignite/metrics/gan/test_fid.py
@@ -110,7 +110,7 @@ def test_wrong_inputs():
         FID(num_features=-1, feature_extractor=torch.nn.Identity())
 
     with pytest.raises(ValueError, match=r"feature_extractor output must be a tensor of dim 2, got: 1"):
-        FID(num_features=1, feature_extractor=torch.nn.Identity()).update(torch.Tensor([[], []]))
+        FID(num_features=1, feature_extractor=torch.nn.Identity()).update(torch.tensor([[], []]))
 
     with pytest.raises(ValueError, match=r"Batch size should be greater than one, got: 0"):
         FID(num_features=1, feature_extractor=torch.nn.Identity()).update(torch.rand(2, 0, 0))
diff --git a/tests/ignite/metrics/test_loss.py b/tests/ignite/metrics/test_loss.py
@@ -31,7 +31,7 @@ def update(self, output):
 
 
 def test_output_as_mapping_without_criterion_kwargs():
-    y_pred = torch.Tensor([[2.0], [-2.0]])
+    y_pred = torch.tensor([[2.0], [-2.0]])
     y = torch.zeros(2)
     criterion_kwargs = {}
 
@@ -42,7 +42,7 @@ def test_output_as_mapping_without_criterion_kwargs():
 
 
 def test_output_as_mapping_with_criterion_kwargs():
-    y_pred = torch.Tensor([[2.0], [-2.0]])
+    y_pred = torch.tensor([[2.0], [-2.0]])
     y = torch.zeros(2)
     criterion_kwargs = {"reduction": "sum"}
 
diff --git a/tests/ignite/metrics/test_mean_pairwise_distance.py b/tests/ignite/metrics/test_mean_pairwise_distance.py
@@ -117,7 +117,7 @@ def _test_distrib_accumulator_device(device):
         for dev in [mpd._device, mpd._sum_of_distances.device]:
             assert dev == metric_device, f"{type(dev)}:{dev} vs {type(metric_device)}:{metric_device}"
 
-        y_pred = torch.Tensor([[3.0, 4.0], [-3.0, -4.0]])
+        y_pred = torch.tensor([[3.0, 4.0], [-3.0, -4.0]])
         y = torch.zeros(2, 2)
         mpd.update((y_pred, y))
 
diff --git a/tests/ignite/metrics/test_metric.py b/tests/ignite/metrics/test_metric.py
