autogluon: RuntimeError: The best config {'search_space▁optimization.lr': 5.5e-05} is not found in config history = OrderedDict().

Using SageMaker, tried conda_python3 (and downloaded mxnet as suggested in https://autogluon.mxnet.io/install.html#installation-faq) and conda_mxnet_36, error is reproduced in both kernels.

Just following the text prediction quick start example, but running into following:

!pip install --upgrade pip
!pip install --upgrade 'scikit-learn<0.23,>=0.22.0'
!pip install --upgrade mxnet autogluon

from autogluon import TextPrediction as task
from autogluon.utils.tabular.utils.loaders.load_pd import load
import pandas as pd
import numpy as np
train_data = load('https://autogluon-text.s3-accelerate.amazonaws.com/glue/sst/train.parquet')
dev_data = load('https://autogluon-text.s3-accelerate.amazonaws.com/glue/sst/dev.parquet')
rand_idx = np.random.permutation(np.arange(len(train_data)))[:2000]
train_data = train_data.iloc[rand_idx]
train_data.head(10)

predictor = task.fit(train_data, label='label',
                     time_limits=60,
                     seed=123,
                     output_directory='./ag_sst')

(No gpu configuration as my instance does not have gpu) Running into: RuntimeError: The best config {'search_space▁optimization.lr': 5.5e-05} is not found in config history = OrderedDict(). This should never happen! error

Edit: tried on both 0.0.13 and 0.0.14, same error. I did check the setup.py and verified that all packages were installed. Not sure if there’s a version mismatch?