Issue with Saving Model in PyMC Marketing (TimedeltaIndex Error)

[stdp95]

Hi everyone,

I’m encountering an issue when trying to save a PyMC Marketing Model in Databricks. Here’s the context:
I successfully ran the model, and when I attempt to save it using the .save() method, I get the following error:
AttributeError: 'TimedeltaIndex' object has no attribute 'unit'File <command-2594243823868214>, line 1 ----> 1 mmm.save("model.nc")
The stack trace indicates an issue in xarray while attempting to use to_netcdf. Specifically, it seems related to the handling of a TimedeltaIndex in time_deltas.unit.

Additional Details:
Environment: Running the code on Databricks.
PyMC-Marketing Version: pymc_marketing-0.10.0.

Has anyone encountered this error or knows how to resolve it?

Thank you for any insights!

[williambdean]

Hi. Could you share the full traceback? The save method is a light wrapper around arviz save function, is it possible to save any other way?

[williambdean]

Hi, @stdp95. Any luck with this?

[stdp95]

No, I don't have a solution for this. Since my training time is less I am re-running the code and using the mmm object for the other parts like budget optimization.

[williambdean]

To be clear, the date should be a column and not an index of the DataFrame.

Are you able to provide a reproducible example for this?

[stdp95]

I don't have any index on the dataframe. Interestingly, when I ran the same code again, the save method worked and I was able to load it without any issues. I suspect that a dependency library might have caused the issue earlier, and it has likely been fixed since then. Thanks for checking on this.

[williambdean]

Happy it worked itself out!