Skip to content

Commit 0620c55

Browse files
committed
make converter default matching v0.3.0
1 parent 6dbd335 commit 0620c55

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

chgnet/graph/converter.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class CrystalGraphConverter(nn.Module):
3333

3434
def __init__(
3535
self,
36-
atom_graph_cutoff: float = 5,
36+
atom_graph_cutoff: float = 6,
3737
bond_graph_cutoff: float = 3,
3838
algorithm: Literal["legacy", "fast"] = "fast",
3939
on_isolated_atoms: Literal["ignore", "warn", "error"] = "error",

chgnet/graph/graph.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def __repr__(self):
5252
def __hash__(self) -> int:
5353
"""Hash this edge."""
5454
img = (self.info or {}).get("image")
55-
img_str = "" if img is None else img.tostring()
55+
img_str = "" if img is None else img.tobytes()
5656
return hash((self.nodes[0], self.nodes[1], img_str))
5757

5858
@abstractmethod

0 commit comments

Comments
 (0)