from  numba import njit
import numpy as np
import math
from numba import config
from numba.experimental.remarks import RemarksInterface


@njit()
def foo(n):
    ret = np.empty(n, dtype=np.float64)
    arr = np.arange(n)
    for x in range(n):
        ret[x] = math.sin(arr[x])
    return ret

with RemarksInterface() as remarks:
    foo(10000)

print(remarks.file_path)

# All nodes without a caller
print(remarks.head_nodes)

# Head node [1] is foo from the above example
for x in remarks.head_nodes[1].remarks:
    args = x.pop("Args") # Remove the Args key, args still needs to be processed properly
    print(x)

    remark_info = ""
    for arg in args:
        for value in arg.values():
            remark_info  = str(value)   " "

    print("Remark String:", remark_info)
    print("\n")
    break # To prevent long outputs