def test_simple_example(): acc = simple_example() assert acc >= 4 / 7
def test_torch_tree_nn_simple_example(initial_embedding, separate_y): torch_tree_nn.simple_example(initial_embedding, separate_y)
def test_torch_tree_nn_simple_example(initial_embedding): torch_tree_nn.simple_example(initial_embedding)