fix gnn imports

cog-imperial · May 8, 2024 · b1a0c44 · b1a0c44
1 parent d34bca5
commit b1a0c44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/omlt/io/torch_geometric/build_gnn_formulation.py b/src/omlt/io/torch_geometric/build_gnn_formulation.py
@@ -1,7 +1,7 @@
 import numpy as np
 import pyomo.environ as pyo
 
-from omlt.io.torch_geometric import load_torch_geometric_sequential
+from omlt.io.torch_geometric.torch_geometric_reader import load_torch_geometric_sequential
 from omlt.neuralnet import FullSpaceNNFormulation