ImportError: cannot import name 'GraphGymDataModule' from 'torch_geometric.graphgym.train'

这个错误通常是因为你使用的 `torch_geometric` 版本太老，或者是因为你的环境中缺少某些依赖包。

首先，请确保你安装了最新版本的 `torch_geometric`，可以使用以下命令升级：

pip install --upgrade torch-geometric

如果更新了 `torch_geometric` 但是仍然出现错误，那么可能是缺少依赖包。请确保你的环境中安装了以下依赖包：

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.9.0+${CUDA}.html

其中 `${CUDA}` 是你的 CUDA 版本号，例如 `cu102` 或 `cu111`。

如果你的环境中已经安装了所有依赖包，那么可能是因为你使用的是 `torch_geometric` 的旧版本，而你的代码依赖于新版本中的某些特性。你可以尝试升级 `torch_geometric` 到最新版，或者修改代码以适配你当前的版本。