避免进口`进口火炬-几何'上的`LLM'树下(和熊猫)
作者: frazane创建于 2026年8月16日更新于 2026年8月16日
建议的重构 import torch_geometric 会引入 LLM 子树 - 以及安装时的 pandas - 因为 torch_geometric/__init__.py 会热情地导入 torch_geometric.datasets,两个数据集模块在模块级别导入了 LLM:datasets/web_qsp_dataset.py 和 datasets/molecule_gpt_dataset.py。在 2.7.0 版本上测试,这会将系统的热导入时间延长约 0.15 秒,其中大部分是 pandas;在网络文件系统较冷的集群中,时间会明显更长(最多几秒钟)。这些导入仍然存在于 master 中。
内容来源: pyg-team/pytorch_geometric