File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -306,7 +306,7 @@ def forward(self, x):
306
306
# Target is PV, so only take the history
307
307
# Copy batch
308
308
x_tmp = x .copy ()
309
- x_tmp [BatchKey .pv ] = x_tmp [BatchKey .pv ][:, : self .history_len + 1 ]
309
+ x_tmp [BatchKey .pv ] = x_tmp [BatchKey .pv ][:, : self .history_len + 1 ]
310
310
modes ["pv" ] = self .pv_encoder (x_tmp )
311
311
312
312
# *********************** GSP Data ************************************
@@ -329,7 +329,7 @@ def forward(self, x):
329
329
else :
330
330
# Have to be its own Batch format
331
331
x_tmp = x .copy ()
332
- x_tmp [BatchKey .wind ] = x_tmp [BatchKey .wind ][:, : self .history_len + 1 ]
332
+ x_tmp [BatchKey .wind ] = x_tmp [BatchKey .wind ][:, : self .history_len + 1 ]
333
333
# This needs to be a Batch as input
334
334
modes ["wind" ] = self .wind_encoder (x_tmp )
335
335
@@ -339,7 +339,7 @@ def forward(self, x):
339
339
modes ["sensor" ] = self .sensor_encoder (x )
340
340
else :
341
341
x_tmp = x .copy ()
342
- x_tmp [BatchKey .sensor ] = x_tmp [BatchKey .sensor ][:, : self .history_len + 1 ]
342
+ x_tmp [BatchKey .sensor ] = x_tmp [BatchKey .sensor ][:, : self .history_len + 1 ]
343
343
# This needs to be a Batch as input
344
344
modes ["sensor" ] = self .sensor_encoder (x_tmp )
345
345
You can’t perform that action at this time.
0 commit comments