diff --git a/dvc/repo/experiments/queue/base.py b/dvc/repo/experiments/queue/base.py index 0312b08e4d..96f19e53f2 100644 --- a/dvc/repo/experiments/queue/base.py +++ b/dvc/repo/experiments/queue/base.py @@ -90,7 +90,7 @@ def __init__(self, repo: "Repo", ref: str, failed_ref: Optional[str] = None): """Construct a queue. Arguments: - scm: Git SCM instance for this queue. + repo: DVC repo instance for this queue. ref: Git stash ref for this queue. failed_ref: Failed run Git stash ref for this queue. """ diff --git a/dvc/repo/experiments/queue/tasks.py b/dvc/repo/experiments/queue/tasks.py index deb250f82d..c88c83e9be 100644 --- a/dvc/repo/experiments/queue/tasks.py +++ b/dvc/repo/experiments/queue/tasks.py @@ -84,8 +84,8 @@ def cleanup_exp(executor: TempDirExecutor, infofile: str) -> None: """Cleanup after an experiment. Arguments: - tmp_dir: Temp directory to be removed. - entry_dict: Serialized QueueEntry for this experiment. + executor: Executor to clean up. + infofile: Path to the executor info file. """ executor.cleanup(infofile)