From 6681aedd017609e6c5aedfdb85ff0c671174ee58 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Fri, 3 Feb 2023 09:36:44 +0000 Subject: [PATCH] add P_LONELY_CHILD_EXTENSION to workflow vars --- .github/workflows/deploy-to-node.yaml | 1 + ansible/deploy-to-node.yaml | 3 +++ backend/oasst_backend/config.py | 2 +- 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/.github/workflows/deploy-to-node.yaml b/.github/workflows/deploy-to-node.yaml index a33966af..3457293e 100644 --- a/.github/workflows/deploy-to-node.yaml +++ b/.github/workflows/deploy-to-node.yaml @@ -41,6 +41,7 @@ jobs: MAX_TREE_DEPTH: ${{ vars.MAX_TREE_DEPTH }} MAX_CHILDREN_COUNT: ${{ vars.MAX_CHILDREN_COUNT }} LONELY_CHILDREN_COUNT: ${{ vars.LONELY_CHILDREN_COUNT }} + P_LONELY_CHILD_EXTENSION: ${{ vars.P_LONELY_CHILD_EXTENSION }} P_ACTIVATE_BACKLOG_TREE: ${{ vars.P_ACTIVATE_BACKLOG_TREE }} MIN_ACTIVE_RANKINGS_PER_LANG: ${{ vars.MIN_ACTIVE_RANKINGS_PER_LANG }} GOAL_TREE_SIZE: ${{ vars.GOAL_TREE_SIZE }} diff --git a/ansible/deploy-to-node.yaml b/ansible/deploy-to-node.yaml index 5c518a5c..23bc0573 100644 --- a/ansible/deploy-to-node.yaml +++ b/ansible/deploy-to-node.yaml @@ -125,6 +125,9 @@ TREE_MANAGER__LONELY_CHILDREN_COUNT: "{{ lookup('ansible.builtin.env', 'LONELY_CHILDREN_COUNT') | default('2', true) }}" + TREE_MANAGER__P_LONELY_CHILD_EXTENSION: + "{{ lookup('ansible.builtin.env', 'P_LONELY_CHILD_EXTENSION') | + default('0.75', true) }}" TREE_MANAGER__P_ACTIVATE_BACKLOG_TREE: "{{ lookup('ansible.builtin.env', 'P_ACTIVATE_BACKLOG_TREE') | default('0.1', true) }}" diff --git a/backend/oasst_backend/config.py b/backend/oasst_backend/config.py index 71d74c29..f1685221 100644 --- a/backend/oasst_backend/config.py +++ b/backend/oasst_backend/config.py @@ -115,7 +115,7 @@ class TreeManagerConfiguration(BaseModel): lonely_children_count: int = 2 """Number of children below which parents are preferred during sampling for reply tasks.""" - p_lonely_child_extension: float = 0.8 + p_lonely_child_extension: float = 0.75 """Probability to select a prompter message parent with less than lonely_children_count children.""" recent_tasks_span_sec: int = 3 * 60 # 3 min