Pickle Error with LGT MAP

[edwinnglabs]

Describe the bug
I run a basic lgt training and try to pickle it. I got the error message below. I am running this under orbit 1.0.10.

To Reproduce

from orbit.models.lgt
import LGTMAP
import pickle
lgt = LGTMAP(**args_dict)
lgt.fit(data1)
with open(MODEL_FILE, 'wb') as f:
    pickle.dump(lgt, f)

Expected behavior
Successful pickle dump.

Screenshots

---

AttributeError Traceback (most recent call last)
in
1 with open(MODEL_FILE, 'wb') as f:
----> 2 pickle.dump(lgt, f)

AttributeError: Can't pickle local object 'BaseLGT._set_init_values..'

Environment (please complete the following information):

• OS: macOS
• Python Version: 3.7.7

Dependencies

Orbit version: 1.0.10
Pandas: 1.2.1
Numpy: 1.19.5