Expected Behavior
The QuantConnect/LEAN Python environment should provide dgl==2.4.0 instead of dgl==2.1.0.
This would improve compatibility with the current PyTorch stack and remove the need for torchdata compatibility workarounds. DGL 2.4.0 release notes state that GraphBolt no longer depends on the deprecated torchdata package, and the release adds support for a newer PyTorch/CUDA stack.
Actual Behavior
The current environment provides dgl==2.1.0. DGL 2.1.0 has limited PyTorch compatibility; its release notes list supported PyTorch versions as 2.0.0/1, 2.1.0/1/2, and 2.2.0/1.
This creates compatibility issues with newer Torch versions, including Torch 2.8, and DGL 2.1.0 still requires torchdata support for GraphBolt-related imports. This makes DGL harder to use in the current QuantConnect Python environment.
Potential Solution
Upgrade the bundled DGL package:
dgl==2.1.0 -> dgl==2.4.0
This should reduce dependency conflicts around torchdata and provide better alignment with the current PyTorch package environment.
Checklist
Expected Behavior
The QuantConnect/LEAN Python environment should provide
dgl==2.4.0instead ofdgl==2.1.0.This would improve compatibility with the current PyTorch stack and remove the need for
torchdatacompatibility workarounds. DGL 2.4.0 release notes state that GraphBolt no longer depends on the deprecatedtorchdatapackage, and the release adds support for a newer PyTorch/CUDA stack.Actual Behavior
The current environment provides
dgl==2.1.0. DGL 2.1.0 has limited PyTorch compatibility; its release notes list supported PyTorch versions as 2.0.0/1, 2.1.0/1/2, and 2.2.0/1.This creates compatibility issues with newer Torch versions, including Torch 2.8, and DGL 2.1.0 still requires
torchdatasupport for GraphBolt-related imports. This makes DGL harder to use in the current QuantConnect Python environment.Potential Solution
Upgrade the bundled DGL package:
dgl==2.1.0->dgl==2.4.0This should reduce dependency conflicts around
torchdataand provide better alignment with the current PyTorch package environment.Checklist
masterbranch