AttributeError: module 'datasets' has no attribute 'Split'

[taqta]

Hi, thanks for your great work. I tried to run create_embeddings.py, but get the error below:

Traceback (most recent call last):
  File "/home/dsphaoyang/centroid/inference/create_embeddings.py", line 13, in <module>
    from train_ctl_model import CTLModel
  File "/home/dsphaoyang/centroid/./train_ctl_model.py", line 13, in <module>
    import pytorch_lightning as pl
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/pytorch_lightning/__init__.py", line 65, in <module>
    from pytorch_lightning.trainer import Trainer
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/pytorch_lightning/trainer/__init__.py", line 18, in <module>
    from pytorch_lightning.trainer.trainer import Trainer
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/pytorch_lightning/trainer/trainer.py", line 32, in <module>
    from pytorch_lightning.loggers import LightningLoggerBase
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/pytorch_lightning/loggers/__init__.py", line 28, in <module>
    from pytorch_lightning.loggers.mlflow import _MLFLOW_AVAILABLE, MLFlowLogger
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/pytorch_lightning/loggers/mlflow.py", line 35, in <module>
    import mlflow
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/__init__.py", line 41, in <module>
    from mlflow import projects  # pylint: disable=unused-import
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/projects/__init__.py", line 9, in <module>
    import mlflow.projects.databricks
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/projects/databricks.py", line 11, in <module>
    from mlflow import tracking
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/tracking/__init__.py", line 20, in <module>
    from mlflow.tracking.fluent import _EXPERIMENT_ID_ENV_VAR, _EXPERIMENT_NAME_ENV_VAR, _RUN_ID_ENV_VAR
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/tracking/fluent.py", line 13, in <module>
    from mlflow.data.dataset import Dataset
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/data/__init__.py", line 5, in <module>
    from mlflow.data import dataset_registry
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/data/dataset_registry.py", line 137, in <module>
    from mlflow.data.huggingface_dataset import from_huggingface
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/data/huggingface_dataset.py", line 11, in <module>
    from mlflow.data.huggingface_dataset_source import HuggingFaceDatasetSource
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/data/huggingface_dataset_source.py", line 10, in <module>
    class HuggingFaceDatasetSource(DatasetSource):
  File "/home/dsphaoyang/anaconda3/envs/centroid/lib/python3.9/site-packages/mlflow/data/huggingface_dataset_source.py", line 23, in HuggingFaceDatasetSource
    split: Optional[Union[str, datasets.Split]] = None,
AttributeError: module 'datasets' has no attribute 'Split'

Does anyone know what the issue might be? Thank you!