%mprun продолжает давать имя ошибки (проблема с импортированными модулями)

Question

Привет я в настоящее время пытаюсь выполнить построчный профиль памяти для одной из моих функций. Однако всякий раз, когда функции требуется импортированный модуль, команда%mprun не работает. Это упрощенный пример. Я пробовал это как на Jupyter, командной строке и Anaconda.

#Here I write a py file containing the function I want to profile
%%file mpruntest.py
@profile
import pandas as pd
import numpy as np

def test(a):
    df = pd.DataFrame(np.random.randn(10,5))
    return df

Это даст мне файл py под названием mpruntest.

#Here I run the mprun
from mpruntest import test
%mprun -f test test(1)

Я продолжаю получать NameError, заявляя, что pd или np не определены. Я не уверен, как мне поступить так, чтобы профилировщик памяти мог работать и считывать, что соответствующие модули импортированы.