We're upgrading our ML jobs from using Runtime 9.1 LTS ML to Runtime 10.4 LTS ML in Databricks. One of the libraries our jobs relying on is Prophet. From 9.1 to 10.4, both the versions of Prophet (1.0.1) and PyStan (2.19.1.1) haven't changed, however, Prophet in 10.4 LTS ML doesn't work properly. Is that possible to get the LTS runtime fixed for Prophet/PyStan or other recommendation in addressing the issue from our end? Many thanks!
Minimal example for testing:
import prophet
import pandas as pd
df = pd.DataFrame({"ds": [1.0, 2.0, 3.0, 4.0], "y": [0.1, 0.2, 0.3, 0.4]})
model = prophet.Prophet()
model.fit(df)
The example works with Runtime 9.1 LTS ML, however not with Runtime 10.4 LTS ML (error message --
AttributeError: 'StanModel' object has no attribute 'fit_class').
