OLS bug #1000

Fix the inappropriate  loading the dataset for the estimations of the transfer opterator using OLS.

Author: Danfoa

nn/DeepProjections.py

@@ -231,7 +231,7 @@ def approximate_transfer_operator(self, train_data_loader: DataLoader):
                 if key not in train_data:
                     train_data[key] = torch.squeeze(value)
                 else:
-                    torch.cat([train_data[key], torch.squeeze(value)], dim=0)
+                    train_data[key] = torch.cat([train_data[key], torch.squeeze(value)], dim=0)
 
         # Apply any pre-processing to the state and next state
         state, next_state = train_data["state"], train_data["next_state"]