From 3b285e2f263b46d7483a78fce2c6fa603a54b032 Mon Sep 17 00:00:00 2001 From: Akshay Kashyap Date: Wed, 8 May 2024 10:41:00 -0400 Subject: [PATCH] nit: docstring --- 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 8005b33d..3cf0516e 100644 --- a/lerobot/common/policies/diffusion/modeling_diffusion.py +++ b/lerobot/common/policies/diffusion/modeling_diffusion.py @@ -30,7 +30,7 @@ from lerobot.common.policies.utils import ( def _make_noise_scheduler(name: str, **kwargs: dict) -> DDPMScheduler | DDIMScheduler: """ - Factory for noise scheduler instances of the given name. All kwargs are passed + Factory for noise scheduler instances of the requested type. All kwargs are passed to the scheduler. """ if name == "DDPM":