AttributeError: Can't get attribute 'xx' on <module '__main__' from 'path/to/kernprof'>

[SysuJayce]

I want to profile method in class with kernprof, but it seems that pickle is not supported. Any help is appreciated.

kernprof version: 4.0.1

Below is the code to reproduce the issue.

import pickle

class Data:
  pass

class Test:
  @profile
  def a(self):
    data = Data()
    pickle.dump(data, open("tmp.pkl", "wb"))

Test().a()

If we save the code into test.py and call kernprof -lv test.py, the error is:

_pickle.PicklingError: Can't pickle <class '__main__.Data'>: attribute lookup Data on __main__ failed

It seems that neither pickle.dump() nor pickle.load() is supporrted by line_profile currently.

[Erotemic]

Does this work with the older 3.x branch?

[Mogost]

Probably related to #191 (actually duplicates #183)

[kinghuang]

I'm having the same problem where using line_profiler appears to interfere with pickling objects. Not sure how to get around this.