small update

Huy Vu2 · Huy Vu2 · commit cf8b1f48265e · 2025-12-29T20:27:19.000-08:00
diff --git a/dfm/src/megatron/data/common/diffusion_energon_datamodule.py b/dfm/src/megatron/data/common/diffusion_energon_datamodule.py
@@ -122,7 +122,7 @@ def __init__(
             num_workers=num_workers,
             packing_buffer_size=packing_buffer_size,
             pin_memory=pin_memory,
-            task_encoder=task_encoder,   
+            task_encoder=task_encoder,
         )
         self.use_train_split_for_val = use_train_split_for_val
 

Original file line number	Diff line number	Diff line change
`@@ -122,7 +122,7 @@ def __init__(`
`122`	`122`	`num_workers=num_workers,`
`123`	`123`	`packing_buffer_size=packing_buffer_size,`
`124`	`124`	`pin_memory=pin_memory,`
`125`		`- task_encoder=task_encoder,`
	`125`	`+ task_encoder=task_encoder,`
`126`	`126`	`)`
`127`	`127`	`self.use_train_split_for_val = use_train_split_for_val`
`128`	`128`