pythoncomplete в энергии - hardcode фабрика функционируют возвраты?

Я использую pythoncomplete omnicompletion в энергии. Это работает отлично, когда я инстанцирую классов непосредственно, например,

import numpy as np
x = np.ndarray(l)

затем x приписывает завершенный правильно.

Но я работаю с numpy и matplotlib, таким образом, обычно используют функции фабрики т.е.

x = np.zeros((2,2)) 
f = plt.figure()  
ax = f.add_subplot(111)

Есть ли любой способ, которым я могу трудно кодировать типы возврата этих общих функций, таким образом, я могу завершиться на возвращенном объекте. (т.е. настроенный некоторые отображения это matplotlib.pyplot.figure возвращает matplotlib.figure. Иллюстрация, np.zeros возвращает np.ndarray и т.д.?