diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index d682ae8aab675..cb98c660d2dee 100644 --- a/tests/trainer/test_trainer.py +++ b/tests/trainer/test_trainer.py @@ -1690,8 +1690,6 @@ def test_predict_return_predictions_cpu(return_predictions, precision, tmpdir): assert len(preds) == 1 assert preds[0].shape == torch.Size([1, 2]) assert preds[0].dtype == (torch.float64 if precision == 64 else torch.float32) - else: - assert preds == 1 @pytest.mark.parametrize(