Update tests/ignite/contrib/metrics/test_roc_curve.py

pytorch · Jul 7, 2023 · e0ab81b · e0ab81b
1 parent 27d64fd
commit e0ab81b
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/tests/ignite/contrib/metrics/test_roc_curve.py b/tests/ignite/contrib/metrics/test_roc_curve.py
@@ -77,8 +77,6 @@ def update_fn(engine, batch):
     roc_curve_metric.attach(engine, "roc_curve")
 
     data = list(range(size // batch_size))
-    fpr, tpr, thresholds = engine.run(data, max_epochs=1).metrics["roc_curve"]
-    fpr, tpr, thresholds = engine.run(data, max_epochs=1).metrics["roc_curve"]
 
     assert np.array_equal(fpr, sk_fpr)
     assert np.array_equal(tpr, sk_tpr)
@@ -162,11 +160,6 @@ def update(engine, i):
 
     fpr, tpr, thresholds = engine.state.metrics["roc_curve"]
 
-    assert isinstance(fpr, torch.Tensor) and fpr.device == device
-    assert isinstance(tpr, torch.Tensor) and tpr.device == device
-    assert isinstance(thresholds, torch.Tensor) and thresholds.device == device
-    fpr, tpr, thresholds = engine.state.metrics["roc_curve"]
-
     assert isinstance(fpr, torch.Tensor) and fpr.device == device
     assert isinstance(tpr, torch.Tensor) and tpr.device == device
     assert isinstance(thresholds, torch.Tensor) and thresholds.device == device