Skip to content

Library Upgrade: dgl #9499

@Nitasurin

Description

@Nitasurin

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

  • I have completely filled out this template
  • I have confirmed that this issue exists on the current master branch
  • I have confirmed that this is not a duplicate issue by searching

Metadata

Metadata

Assignees

No one assigned

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions