KeyError: 'autormok' on loading

[aiv1]

I've saved and loaded trained model as in tutorial. Got following error:

KeyError Traceback (most recent call last)
Cell In[51], line 1
----> 1 nf2 = NeuralForecast.load(path='./checkpoints/test_run/')
2 Y_hat_df = nf2.predict().reset_index()
3 Y_hat_df.head()

File c:\Users\mk\micromamba\envs\neuralforecast\Lib\site-packages\neuralforecast\core.py:1507, in NeuralForecast.load(path, verbose, **kwargs)
1505 model_name = "".join(model.split("")[:-1])
1506 model_class_name = alias_to_model.get(model_name, model_name)
-> 1507 loaded_model = MODEL_FILENAME_DICT[model_class_name].load(
1508 f"{path}/{model}", **kwargs
1509 )
1510 loaded_model.alias = model_name
1511 models.append(loaded_model)

KeyError: 'autormok'

[elephaint]

Ah, we forgot to include a piece of code, I've issued a PR for this, tjhanks for reporting.

[elephaint]

#1167 Fixes this