From 0369b2804be2fb432b7bae4c9dcea2ddd421b90d Mon Sep 17 00:00:00 2001 From: Ilya Burakov Date: Thu, 31 Aug 2023 16:14:20 +0000 Subject: [PATCH] Fix node conversion tests --- ml/synthesis/src/tests/fixtures/test_tree_root.py | 2 +- ml/synthesis/src/tests/test_node_json_conversion.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/ml/synthesis/src/tests/fixtures/test_tree_root.py b/ml/synthesis/src/tests/fixtures/test_tree_root.py index cbadfa25d..f4af9a133 100644 --- a/ml/synthesis/src/tests/fixtures/test_tree_root.py +++ b/ml/synthesis/src/tests/fixtures/test_tree_root.py @@ -20,7 +20,7 @@ def test_root(): is_terminal=True, is_finish=True, parameters={"a": 1, "b": 2}, - decision=dict(tag="BindDecisionView"), + decision=dict(tag="SingleBindView"), duration=None, sid="-0", children=[], diff --git a/ml/synthesis/src/tests/test_node_json_conversion.py b/ml/synthesis/src/tests/test_node_json_conversion.py index 414d09d3c..21488c9d7 100644 --- a/ml/synthesis/src/tests/test_node_json_conversion.py +++ b/ml/synthesis/src/tests/test_node_json_conversion.py @@ -13,9 +13,7 @@ def test_node_from_raw_dict(test_root, test_root_dict): def test_node_to_train_df_dict(test_root: NittaNodeInTree): assert test_root.children, "test root must have children for this test to run" assert nitta_node_to_df_dict(test_root, siblings=test_root.children, example="test") == { - "alt_bindings": 1, - "alt_dataflows": 0, - "alt_refactorings": 0, + "siblings_SingleBindView": 1, "example": "test", "is_terminal": False, "old_score": 1010,