Fix policy defaults (#113)

2024-04-29 08:26:59 +01:00
parent 791506dfb8
commit ccffa9e406
3 changed files with 13 additions and 2 deletions
--- a/lerobot/common/policies/diffusion/modeling_diffusion.py
+++ b/lerobot/common/policies/diffusion/modeling_diffusion.py
@@ -43,7 +43,7 @@ class DiffusionPolicy(nn.Module):
    name = "diffusion"

    def __init__(
-        self, cfg: DiffusionConfig | None, lr_scheduler_num_training_steps: int = 0, dataset_stats=None
+        self, cfg: DiffusionConfig | None = None, lr_scheduler_num_training_steps: int = 0, dataset_stats=None
    ):
        """
        Args: