Apply suggestions from code review

Co-authored-by: Tal Ben-Nun <[email protected]>
LLNL · Mar 26, 2023 · 154186c · 154186c
1 parent 14c86ed
commit 154186c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/applications/graph/MeshGraphNet/README.md b/applications/graph/MeshGraphNet/README.md
@@ -1,7 +1,8 @@
 ## Mesh Graph Networks
 
-This example contains LBANN implementation of mesh-based graph neural network with 
-syntheticly generated data. 
+This example contains an LBANN implementation of mesh-based graph neural network (MeshGraphNet) with 
+synthetically generated data. 
+For more information about the model, refer to: T. Pfaff et al., "Learning Mesh-Based Simulation with Graph Networks". ICLR'21.
 
 ---
 ### Running the example

diff --git a/applications/graph/MeshGraphNet/SyntheticData.py b/applications/graph/MeshGraphNet/SyntheticData.py
@@ -34,5 +34,3 @@ def sample_dims_func():
   return (size, )
 
 
-if __name__ == '__main__':
-    print(NUM_NODES)