From 7461a9131d91a2c37c412418eb83d2ffe648247e Mon Sep 17 00:00:00 2001 From: Akshay Kashyap Date: Wed, 8 May 2024 10:36:27 -0400 Subject: [PATCH] kwargs fix --- lerobot/common/policies/diffusion/modeling_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/policies/diffusion/modeling_diffusion.py b/lerobot/common/policies/diffusion/modeling_diffusion.py index 4ca21413..8005b33d 100644 --- a/lerobot/common/policies/diffusion/modeling_diffusion.py +++ b/lerobot/common/policies/diffusion/modeling_diffusion.py @@ -28,7 +28,7 @@ from lerobot.common.policies.utils import ( ) -def _make_noise_scheduler(name: str, kwargs: dict) -> DDPMScheduler | DDIMScheduler: +def _make_noise_scheduler(name: str, **kwargs: dict) -> DDPMScheduler | DDIMScheduler: """ Factory for noise scheduler instances of the given name. All kwargs are passed to the scheduler.