From 8edea1dea340af0e39ac4c8d45e2977e1ab5167b Mon Sep 17 00:00:00 2001 From: Alexey Masterov Date: Tue, 22 Jul 2025 13:41:09 +0200 Subject: [PATCH] Do not consider restored branches as leaf ones --- test_runner/random_ops/test_random_ops.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test_runner/random_ops/test_random_ops.py b/test_runner/random_ops/test_random_ops.py index 28f5a18ed8..b85ea16315 100644 --- a/test_runner/random_ops/test_random_ops.py +++ b/test_runner/random_ops/test_random_ops.py @@ -97,6 +97,9 @@ class NeonBranch: self.name: str | None = None if "name" in branch["branch"]: self.name = branch["branch"]["name"] + self.restored_from: str | None = None + if "restored_from" in branch: + self.restored_from = branch["restored_from"] self.project: NeonProject = project self.neon_api: NeonAPI = project.neon_api self.project_id: str = branch["branch"]["project_id"] @@ -351,7 +354,7 @@ class NeonProject: if branch_id not in self.reset_branches: self.terminate_benchmark(branch_id) self.neon_api.delete_branch(self.id, branch_id) - if len(parent.children) == 1 and parent.id != self.main_branch.id: + if len(parent.children) == 1 and parent.id != self.main_branch.id and parent.restored_from is None: self.leaf_branches[parent.id] = parent parent.children.pop(branch_id) if branch_id in self.leaf_branches: