Get 'Can't get attribute" error while loading my pickel file

Question

I am trying to use pick to save and load my ML models but I get an error. Here is the simplify version of my code to save my model:

import pickle
def test(x,y):
    return x+y

filename = 'test.pkl'
pickle.dump(test, open(filename, 'wb'))

I can load the pickle file from the same notebook that I am creating it but if I close the notebook and try to load the pick in a new one with the below code:

import pickle
filename = 'test.pkl'
loaded_model = pickle.load(open(filename, 'rb'))

It gets me this error:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[2], line 2
      1 filename = 'test.pkl'
----> 2 loaded_model = pickle.load(open(filename, 'rb'))

AttributeError: Can't get attribute 'test' on <module '__main__'>

Does this answer your question? Is there an easy way to pickle a python function (or otherwise serialize its code)? — pho
– pho, Commented Jan 10, 2023 at 19:26
That post suggest using other packages which are not compatible with some of the Python ML libraries os I am looking for a solution to be able to use pickle — Amir
– Amir, Commented Jan 10, 2023 at 19:28
So this link suggests evey time I want to load my script/model it will re-run the whole script. Then what is the advantage of pickling the model? — Amir
– Amir, Commented Jan 10, 2023 at 20:44

Letsios Matthaios · Accepted Answer · 2023-07-19 15:26:32Z

1

I had exactly the same issue as you. To fix this you need to import the function test in the file that you open the pickle file. In your example if the test function is located in the file ml_model.py, you need to add line from ml_model import test in the second file.

answered Jul 19, 2023 at 15:26

Letsios Matthaios

3364 silver badges10 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Leo Jan 21 at 13:04

Thanks, this solved my problem! Is this a bug in pickle? seems odd that you need to import the function/class if you don't need to explicitly use it.

Collectives™ on Stack Overflow

Get 'Can't get attribute" error while loading my pickel file

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related