fix: 3.3 tests

Author: pawelpaszki

tests/e2e/mnist.py

@@ -40,7 +40,7 @@
 print("MASTER_PORT: is ", os.getenv("MASTER_PORT"))
 
 print("ACCELERATOR: is ", os.getenv("ACCELERATOR"))
-ACCELERATOR = os.getenv("ACCELERATOR")
+ACCELERATOR = os.getenv("ACCELERATOR", "auto")
 
 # If GPU is requested but CUDA is not available, fall back to CPU
 if ACCELERATOR == "gpu" and not torch.cuda.is_available():
@@ -253,7 +253,7 @@ def test_dataloader(self):
     callbacks=[TQDMProgressBar(refresh_rate=20)],
     num_nodes=int(os.environ.get("GROUP_WORLD_SIZE", 1)),
     devices=int(os.environ.get("LOCAL_WORLD_SIZE", 1)),
-    replace_sampler_ddp=False,
+    use_distributed_sampler=False,
     strategy="ddp",
 )
 

tests/e2e/mnist_pip_requirements.txt

@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
 torch==2.7.1+cu118
 torchvision==0.22.1+cu118
-pytorch_lightning==1.9.5
+pytorch_lightning==2.4.0
 torchmetrics==1.8.2
 minio