Skip to content
This repository was archived by the owner on Jun 2, 2025. It is now read-only.

Commit b5de520

Browse files
committed
Remove dropout on Senosr
1 parent b0195d5 commit b5de520

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

ocf_datapipes/training/common.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -885,6 +885,15 @@ def slice_datapipes_by_time(
885885
fill_selection=production,
886886
)
887887

888+
sensor_dropout_time_datapipe = get_t0_datapipe("sensor").draw_dropout_time(
889+
dropout_timedeltas=0,
890+
dropout_frac=0,
891+
)
892+
893+
datapipes_dict["sensor"] = datapipes_dict["sensor"].apply_dropout_time(
894+
dropout_time_datapipe=sensor_dropout_time_datapipe,
895+
)
896+
888897
if "gsp" in datapipes_dict:
889898
datapipes_dict["gsp"], dp = datapipes_dict["gsp"].fork(2, buffer_size=5)
890899

0 commit comments

Comments
 (0)