changes to show correct progress bar numbers when using max_steps

Author: bandpooja

src/lightning/pytorch/callbacks/progress/progress_bar.py

@@ -85,6 +85,13 @@ def total_train_batches(self) -> Union[int, float]:
         dataloader is of infinite size.
 
         """
+        if (
+            self.trainer.max_epochs == -1
+            and self.trainer.max_steps is not None
+            and self.trainer.max_steps > 0
+        ):
+            remaining_steps = self.trainer.max_steps - self.trainer.global_step
+            return min(self.trainer.num_training_batches, remaining_steps)
         return self.trainer.num_training_batches
 
     @property