Skip to content

Commit 85023f4

Browse files
authored
Update stargraph.py
1 parent 420b747 commit 85023f4

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

data/stargraph.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ def __init__(self, maxNodes):
2525
# stargraph has no eos token, padded sequence masking
2626
# doesn't need to happen, but this property is needed
2727
# for compatibility with the bst training code
28-
self.eos_token_id = maxNodes + 5
29-
28+
self.eos_token_id = maxNodes + 4 # small bug fixed by JaydenTeoh
29+
3030
def encode(self, data):
3131
out = []
3232
i = 0

0 commit comments

Comments
 (0)