Bug Report: assert module.__class__ in self._origin_call

[markson14]

When I run the stat() code, it gives "assert module.class in self._origin_call". I print out the entire log and I find out this module has 3 children and still being assigned to wrap_call

Then I went through the code and I tried to print where each module being assigned to wrap_call function from Module._call_impl.

The following image is the print log. It seems the log cannot catch the place where this module has been assiged to wrap_call

Thus I modified the code to this and this problem can be temporarily solved

Can you help me to find out the problem? Thank you.