Rename _load_model_ensemble -> load_model_ensemble_and_task

Summary: Pull Request resolved: fairinternal/fairseq-py#738 Differential Revision: D16377803 Pulled By: myleott fbshipit-source-id: 6beb2f78e7464b70ff65a965d2b747cdca0ca951
facebookresearch · Jul 19, 2019 · 69d0f7f · 69d0f7f
1 parent 7efde22
commit 69d0f7f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/fairseq/checkpoint_utils.py b/fairseq/checkpoint_utils.py
@@ -153,11 +153,11 @@ def load_model_ensemble(filenames, arg_overrides=None, task=None):
             were used during model training
         task (fairseq.tasks.FairseqTask, optional): task to use for loading
     """
-    ensemble, args, _task = _load_model_ensemble(filenames, arg_overrides, task)
+    ensemble, args, _task = load_model_ensemble_and_task(filenames, arg_overrides, task)
     return ensemble, args
 
 
-def _load_model_ensemble(filenames, arg_overrides=None, task=None):
+def load_model_ensemble_and_task(filenames, arg_overrides=None, task=None):
     from fairseq import tasks
 
     ensemble = []

diff --git a/fairseq/models/fairseq_model.py b/fairseq/models/fairseq_model.py
@@ -191,7 +191,7 @@ def from_pretrained(cls, model_name_or_path, checkpoint_file='model.pt', data_na
             if os.path.exists(path):
                 kwargs[arg] = path
 
-        models, args, task = checkpoint_utils._load_model_ensemble(
+        models, args, task = checkpoint_utils.load_model_ensemble_and_task(
             [os.path.join(model_path, cpt) for cpt in checkpoint_file.split(':')],
             arg_overrides=kwargs,
         )