diff --git a/docs/hydra_integration.md b/docs/hydra_integration.md index 04c797fe..6a152983 100644 --- a/docs/hydra_integration.md +++ b/docs/hydra_integration.md @@ -120,7 +120,7 @@ class LanguageModelingConfig(FairseqDataclass): ... @register_task("language_modeling", dataclass=LanguageModelingConfig) -class LanguageModelingTask(LegacyFairseqTask): +class LanguageModelingTask(FairseqTask): ... @classmethod def setup_task(cls, cfg: LanguageModelingConfig):