File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -169,9 +169,9 @@ jobs:
169169 import torch
170170 from pymatgen.core import Lattice, Structure
171171 from pymatgen.optimization.neighbors import find_points_in_spheres
172- from matgl.apps._pes_pyg import Potential
173- from matgl.ext._lammps import LAMMPSMatGLModel
174- from matgl.models._tensornet_pyg import TensorNet
172+ from matgl.apps.pes import Potential
173+ from matgl.ext.lammps import LAMMPSMatGLModel
174+ from matgl.models._tensornet import TensorNet
175175
176176 torch.manual_seed(0)
177177 m = TensorNet(
@@ -443,9 +443,9 @@ jobs:
443443 import torch
444444 from pymatgen.core import Lattice, Structure
445445 from pymatgen.optimization.neighbors import find_points_in_spheres
446- from matgl.apps._pes_pyg import Potential
447- from matgl.ext._lammps import LAMMPSMatGLModel
448- from matgl.models._tensornet_pyg import TensorNet
446+ from matgl.apps.pes import Potential
447+ from matgl.ext.lammps import LAMMPSMatGLModel
448+ from matgl.models._tensornet import TensorNet
449449
450450 torch.manual_seed(0)
451451 m = TensorNet(
You can’t perform that action at this time.
0 commit comments