ImportError: cannot import name 'BaseChronosPipeline' from 'chronos'
Description:
I am trying to use the Chronos time series model in my Jupyter notebook. I installed the chronos package and attempted to load the pipeline as shown in the Hugging Face documentation:
import chronos
pipeline = chronos.BaseChronosPipeline.from_pretrained(
"amazon/chronos-bolt-tiny",
device_map="cuda", # use "cpu" for CPU inference and "mps" for Apple Silicon
torch_dtype=torch.bfloat16,
)
However, I get the following error:
ImportError: cannot import name 'BaseChronosPipeline' from 'chronos' (c:\Users\rohit\AppData\Local\Programs\Python\Python313\Lib\site-packages\chronos_init_.py)
When I check dir(chronos), I only see the default Python module attributes and not BaseChronosPipeline.
I also tried installing chronos-timeseries but pip could not find the package.
Questions:
Is there a different package I should be installing?
Has the API changed, or is there a different way to load the Chronos pipeline?
What is the correct way to use Chronos for time series forecasting with Hugging Face?
Environment:
Python 3.13.0
Windows 10
Installed chronos via pip
Thank you for your help!