adjust factory comment

This commit is contained in:
Bryson Jones
2025-12-10 11:11:05 -08:00
parent adabb37af6
commit 6f856016c5
+1 -1
View File
@@ -135,7 +135,7 @@ def make_policy_config(policy_type: str, **kwargs) -> PreTrainedConfig:
Args:
policy_type: The type of the policy. Supported types include "tdmpc",
"multi_task_dit", "diffusion", "act", "vqbet", "pi0", "pi05", "sac",
"reward_classifier", "smolvla".
"smolvla", "reward_classifier".
**kwargs: Keyword arguments to be passed to the configuration class constructor.
Returns: