Unpickling a class raises attribute error

Question

I wrote a class A in the Class_A.py file

class A():
    def square(self, num):
        return num*num 

Next in pickle_A.py, I saved the class