drop duplicate test

Author: avantikalal

tests/test_lightning.py

@@ -377,10 +377,6 @@ def test_lightning_model_ensemble():
     preds = model.predict_on_dataset(dataset=udataset, devices="cpu")
     assert preds.shape == (2, 1, 1)
 
-    model.reset_transform()
-    preds = model.predict_on_dataset(dataset=udataset, devices="cpu")
-    assert preds.shape == (2, 4, 1)
-
 
 bin_model = generate_model(task="binary", loss="bce", n_tasks=2)
 bin_model.model_params["crop_len"] = 0