Memory_profiler в блокноте возврата объекта Jupiter недействителен

Question

В настоящее время я пытаюсь использовать memory_profiler в записной книжке Jupiter в качестве следующего кода:

%load_ext memory_profiler

def test1(n):
    return n+1
def test2(n):
    i = test1(n)
    %memit i
    %memit test2(n)
test2(10)

И он возвращает ошибку "NameError: имя 'i' не определено"

Не могли бы вы, ребята, помочь мне найти правильный способ использования memory_profiler в ноутбуке jupyter

Большое спасибо.