pytorch

1
"""
2
Helper function used in test_decorator.py. We define it in a
3
separate file on purpose to test that the names in different modules
4
are resolved correctly.
5
"""
6

7
from jit.mydecorator import my_decorator
8
from jit.myfunction_b import my_function_b
9

10

11
@my_decorator
12
def my_function_a(x: float) -> float:
13
    return my_function_b(x) + 1
14