From 4a875e0a399b4714c75ac9be0d30798a24fb5b6b Mon Sep 17 00:00:00 2001 From: BowenD-UCB <84425382+BowenD-UCB@users.noreply.github.com> Date: Wed, 20 Mar 2024 15:10:18 -0700 Subject: [PATCH] fixed texts --- src/matgl/graph/compute.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/matgl/graph/compute.py b/src/matgl/graph/compute.py index 958d2cb9..9a5645cb 100644 --- a/src/matgl/graph/compute.py +++ b/src/matgl/graph/compute.py @@ -1,4 +1,4 @@ -"""Computing various g based operations.""" +"""Computing various graph based operations.""" from __future__ import annotations @@ -227,7 +227,7 @@ def _create_directed_line_graph(graph: dgl.DGLGraph, threebody_cutoff: float) -> threebody_cutoff: cutoff for three-body interactions Returns: - l_g: DGL line graph of pruned graph to three body cutoff + line_graph: DGL line graph of pruned graph to three body cutoff """ with torch.no_grad(): pg = prune_edges_by_features(graph, feat_name="bond_dist", condition=lambda x: torch.gt(x, threebody_cutoff))