Dependency Issue - Google Colab Notebook

[niti-go]

Running the first cell of the Ludwig Python API interactive notebook found on Ludwig's Text Classification Page gives dependency conflict errors.

To Reproduce:

1. Open the Google Colab notebook: Ludwig Python API Text Classification
2. Restart the runtime if needed. Select T4 GPU.
3. Run the first cell:

# Prerequisite: Installs the latest version of Ludwig in the Colab environment
!pip uninstall -y tensorflow
!python -m pip install git+https://github.com/ludwig-ai/ludwig.git --quiet

4. See error:

Found existing installation: tensorflow 2.18.0
Uninstalling tensorflow-2.18.0:
Successfully uninstalled tensorflow-2.18.0
Installing build dependencies ... done
Getting requirements to build wheel ... done
Preparing metadata (pyproject.toml) ... done
━━━━━━━━━━━━━━━━━━━━━━━━━━━154.1/154.1 kB 12.9 MB/s eta 0:00:00
━━━━━━━━━━━━━━━━━━━━━━━━━━━ 98.1/98.1 kB 8.9 MB/s eta 0:00:00
Installing build dependencies ... done
Getting requirements to build wheel ... done
Preparing metadata (pyproject.toml) ... done
━━━━━━━━━━━━━━━━━━━━━━━━━━━ 50.1/50.1 kB 4.8 MB/s eta 0:00:00
━━━━━━━━━━━━━━━━━━━━━━━━━━━ 61.0/61.0 kB 353.7 kB/s eta 0:00:00
...
━━━━━━━━━━━━━━━━━━━━━━━━━━━ 11.7/11.7 MB 114.2 MB/s eta 0:00:00
Building wheel for ludwig (pyproject.toml) ... done
Building wheel for gpustat (pyproject.toml) ... done

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
tensorflow-decision-forests 1.11.0 requires tensorflow==2.18.0, which is not installed.
dopamine-rl 4.1.2 requires tensorflow>=2.2.0, which is not installed.
google-colab 1.0.0 requires pandas==2.2.2, but you have pandas 2.1.4 which is incompatible.
google-genai 1.18.0 requires pydantic<3.0.0,>=2.0.0, but you have pydantic 1.10.22 which is incompatible.
albumentations 2.0.8 requires pydantic>=2.9.2, but you have pydantic 1.10.22 which is incompatible.
gcsfs 2025.3.2 requires fsspec==2025.3.2, but you have fsspec 2023.10.0 which is incompatible.
gradio 5.31.0 requires pydantic<2.12,>=2.0, but you have pydantic 1.10.22 which is incompatible.
langchain-core 0.3.63 requires pydantic>=2.7.4, but you have pydantic 1.10.22 which is incompatible.
plotnine 0.14.5 requires pandas>=2.2.0, but you have pandas 2.1.4 which is incompatible.
pymc 5.23.0 requires rich>=13.7.1, but you have rich 12.4.4 which is incompatible.
bigframes 2.5.0 requires pyarrow>=15.0.2, but you have pyarrow 14.0.2 which is incompatible.
langchain 0.3.25 requires pydantic<3.0.0,>=2.7.4, but you have pydantic 1.10.22 which is incompatible.
mizani 0.13.5 requires pandas>=2.2.0, but you have pandas 2.1.4 which is incompatible.

Environment:

• OS: \Google Colab (Linux)\
• Version: N/A
• Python version: Python 3.11.13
• Ludwig version: 0.10.4.dev0

Additional context:

Because of this error, I'm also getting an error running the notebook's second cell below. Any guidance is appreciated. Thanks!