Make executor info compatible with old versions.

This commit is contained in:
karajan1001
2022-09-26 16:25:47 +08:00
committed by Gao
parent d8a4ecd68f
commit c2829f4ce8
2 changed files with 16 additions and 0 deletions
+3
View File
@@ -92,6 +92,9 @@ class ExecutorInfo:
@classmethod
def from_dict(cls, d):
if "collected" in d:
if d.pop("collected"):
d["status"] = TaskStatus.FINISHED
return cls(**d)
def asdict(self):
@@ -93,3 +93,16 @@ def test_workspace_executor_failed_status(
else:
executor_info = ExecutorInfo.load_json(infofile)
assert executor_info.status == TaskStatus.FAILED
def test_executor_status_compatiblity():
data = {
"git_url": "file:///Users/home",
"baseline_rev": "123",
"location": "dvc-task",
"root_dir": "/Users/home/8088/.dvc/tmp/exps/tmpx85892cx",
"dvc_dir": ".dvc",
"collected": True,
}
result = ExecutorInfo.from_dict(data)
assert result.status == TaskStatus.FINISHED