Overwrite environment deepcopy

[lt-brs]

No description provided.

[lt-brs]

Related to #82, overwriting deepcopy enabled to deepcopy a bmb.Model

[tomicapretto]

Thanks for the PR!

Do you have an example or test we could add?

[lt-brs]

Yes sure. Here is the test I did.

The point is that it should be included in bambinos.bambi (not in formulae) and it needs the version of formulae including the fix.
In bambinos.formulae, the only relevant test would be to ensure deepcopy(env) don't raise error.

import bambi as bmb
import pandas as pd 
from copy import deepcopy

model = bmb.Model('y ~ x', 
          data=pd.DataFrame({'x':[i for i in range(10)], 'y':[2*i+0.1/(1+i) for i in range(10)]})
          )

trace = model.fit(random_seed=0)

clone_model = deepcopy(model)
clone_trace = clone_model.fit(random_seed=0)

assert clone_trace == trace

Thank you for your message

[tomicapretto]

What happens if you do something like

import bambi as bmb
import numpy as np
import pandas as pd


model = bmb.Model(
    'y ~ np.log(x)', 
    data=pd.DataFrame({'x':[i + 1 for i in range(10)], 'y':[2*i+0.1/(1+i) for i in range(10)]})
)

clone_model = deepcopy(model)
clone_trace = clone_model.fit(random_seed=0)
clone_model.predict(clone_trace, data=pd.DataFrame({"x": [1, 2, 3, 4]})

Does the last predict work?

edit I'm expecting it to fail. Since the environment in the second model is empty (if I understand correctly what's happening with deepcopy) and then it won't be able to find where np.log() comes from.

[lt-brs]

Indeed you are right, the predict fails in this case. Thank you for your help.
I looked at bambinos/bambi#400 and especially the comment of August 24, 2021.
It seems that dill enable to fix the issue. I updated the PR by adding the fix and a dependency to dill

The test you previously mentionned work

[tomicapretto]

Brilliant! Thanks for taking the work to do ti. I'm triggering CI now but I will be checking with more care tomorrow.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 95.55%. Comparing base (d7eae85) to head (716f06c).
⚠️ Report is 41 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master      #83      +/-   ##
==========================================
+ Coverage   95.54%   95.55%   +0.01%     
==========================================
  Files          30       30              
  Lines        3543     3552       +9     
==========================================
+ Hits         3385     3394       +9     
  Misses        158      158              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[tomicapretto]

Great to see the CI is passing!

@lt-brs Could you tell me a little more about your use case? I would like to know more about how you're using this. I understand you want to pickle a Bambi model and be able to load it again in the future. But I don't know if you have more use cases in mind. Also, if something you're testing fails, could you let me know, please?

I'm quite confident this change is not breaking anything of the existing functionalities (all tests are good). But I don't have enough confidence to assume the new feature is working as expected yet, simply because I don't have them in mind right now.