add data conversion to np.float32 for exact parity with standard batching

Author: OckermanSethGVSU

torch_geometric_temporal/dataset/pems_bay.py

@@ -186,7 +186,8 @@ def get_index_dataset(self, lags: int = 12, batch_size: int = 64, shuffle: bool
 
         # setup data
         data = np.load(os.path.join(self.raw_data_dir, "pems_node_values.npy")).transpose((1, 2, 0))
-        
+        data = data.astype(np.float32)
+
         if allGPU != -1:
             data = torch.tensor(data,dtype=torch.float).to(f"cuda:{allGPU}")
             means = torch.mean(data, dim=(0, 2), keepdim=True)