Bug fix in Graph class

2025-02-09 13:01:58 +01:00
parent 86405ef597
commit 7702427e8d
2 changed files with 4 additions and 3 deletions
--- a/pina/init.py
+++ b/pina/init.py
@@ -1,6 +1,7 @@
 __all__ = [
    "Trainer", "LabelTensor", "Plotter", "Condition",
    "PinaDataModule", 'TorchOptimizer', 'Graph',
+    "RadiusGraph", "KNNGraph"
 ]

 from .meta import *
@@ -14,4 +15,4 @@ from .data import PinaDataModule

 from .optim import TorchOptimizer
 from .optim import TorchScheduler
-from .graph import Graph
+from .graph import Graph, RadiusGraph, KNNGraph