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,