Update wgan_gp/training.py

Redundant code removed

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: Roman223

wgan_gp/training.py

@@ -458,7 +458,7 @@ def sample_from_tensor(tensor, n_samples):
         """
         indices = torch.randperm(tensor.size(0))
         indices_trunc = indices[:n_samples]
-        sampled_tensor = tensor[indices_trunc[:n_samples]]
+        sampled_tensor = tensor[indices_trunc]
         return sampled_tensor
 
     def calculate_mfs_torch(