update workflow

Author: ajithvcoder

.github/workflows/01E_Deployment_PR_Train_And_Store.yaml

@@ -210,12 +210,7 @@ jobs:
             -e NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics \
             -e NVIDIA_VISIBLE_DEVICES=all \
             --rm ${{ secrets.AWS_ECR_LOGIN_URI }}/${{ secrets.ECR_REPOSITORY_NAME }}:latest \
-            python -c "
-              import torch;
-              from lightning.pytorch.accelerators import CUDAAccelerator;
-              print(f'PyTorch CUDA: {torch.cuda.is_available()}');
-              print(f'Lightning GPUs: {CUDAAccelerator.auto_device_count()}')
-            "
+            python -c "import torch; from lightning.pytorch.accelerators import CUDAAccelerator; print(f'PyTorch CUDA: {torch.cuda.is_available()}'); print(f'Lightning GPUs: {CUDAAccelerator.auto_device_count()}')"
 
       - name: Run DVC commands in container
         run: |