search

datamol-io / graphium

Graphium: Scaling molecular GNNs to infinity.
https://graphium-docs.datamol.io/
Apache License 2.0
197 stars 12 forks source link

Import error with `lightning_graphcore` #496

Closed DomInvivo closed 9 months ago

DomInvivo commented 9 months ago

There might have been some recent changes with lightning_graphcore which causes the unit-tests to fail here. The code crashes when doing from lightning_graphcore import IPUStrategy.

________________ ERROR collecting tests/test_ipu_dataloader.py _________________
ImportError while importing test module '/home/runner/work/graphium/graphium/tests/test_ipu_dataloader.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
../../../micromamba/envs/graphium/lib/python3.8/importlib/__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
tests/test_ipu_dataloader.py:15: in <module>
    from lightning_graphcore import IPUStrategy
../../../micromamba/envs/graphium/lib/python3.8/site-packages/lightning_graphcore/__init__.py:8: in <module>
    from lightning_graphcore.strategy import IPUStrategy
../../../micromamba/envs/graphium/lib/python3.8/site-packages/lightning_graphcore/strategy.py:58: in <module>
    from lightning_graphcore.utils import _fp_to_half, _LightningModuleWrapperBase
../../../micromamba/envs/graphium/lib/python3.8/site-packages/lightning_graphcore/utils.py:23: in <module>
    from lightning.pytorch.overrides.base import _LightningPrecisionModuleWrapperBase
E   ModuleNotFoundError: No module named 'lightning.pytorch.overrides.base'
DomInvivo commented 9 months ago

The issue seems to be solved. Not sure what happened.