Dev Update (#582)
* Fix adaptive refinement (#571) --------- Co-authored-by: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> * Remove collector * Fixes * Fixes * rm unnecessary comment * fix advection (#581) * Fix tutorial .html link (#580) * fix problem data collection for v0.1 (#584) * Message Passing Module (#516) * add deep tensor network block * add interaction network block * add radial field network block * add schnet block * add equivariant network block * fix + tests + doc files * fix egnn + equivariance/invariance tests Co-authored-by: Dario Coscia <dariocos99@gmail.com> --------- Co-authored-by: giovanni <giovanni.canali98@yahoo.it> Co-authored-by: AleDinve <giuseppealessio.d@student.unisi.it> * add type checker (#527) --------- Co-authored-by: Filippo Olivo <filippo@filippoolivo.com> Co-authored-by: Giovanni Canali <115086358+GiovanniCanali@users.noreply.github.com> Co-authored-by: giovanni <giovanni.canali98@yahoo.it> Co-authored-by: AleDinve <giuseppealessio.d@student.unisi.it>
This commit is contained in:
13
pina/model/block/message_passing/__init__.py
Normal file
13
pina/model/block/message_passing/__init__.py
Normal file
@@ -0,0 +1,13 @@
|
||||
"""Module for the message passing blocks of the graph neural models."""
|
||||
|
||||
__all__ = [
|
||||
"InteractionNetworkBlock",
|
||||
"DeepTensorNetworkBlock",
|
||||
"EnEquivariantNetworkBlock",
|
||||
"RadialFieldNetworkBlock",
|
||||
]
|
||||
|
||||
from .interaction_network_block import InteractionNetworkBlock
|
||||
from .deep_tensor_network_block import DeepTensorNetworkBlock
|
||||
from .en_equivariant_network_block import EnEquivariantNetworkBlock
|
||||
from .radial_field_network_block import RadialFieldNetworkBlock
|
||||
Reference in New Issue
Block a user