Hidden states in TermEncoder accessed before initialized

[vrjmath]

In the TermEncoder class, the hidden_states dictionary is declared at line 148
hidden_states = {} # node -> hidden state,
accessed at line 168
h = hidden_states[c],
and initialized at line 185
hidden_states[node] = hiddens[i].
I seem to be getting at error because hidden_states is empty when the code tries to access it at line 168.

[yangky11]

Thanks for your interest in our work!

In each iteration, we access the nodes of a specific height. And we iterate from height 0 to 1, 2, ... etc.

CoqGym/ASTactic/models/term_encoder.py

Line 152 in 05fa457

for height in sorted(height2nodes.keys()):

Nodes with height 0 are leaf nodes. They do not have children. So this inner loop will not be executed in the first iteration of the outer loop:

CoqGym/ASTactic/models/term_encoder.py

Line 16 in 05fa457

'constructor_rel',

Therefore, hidden_states should have been initialized when it is accessed. For each node, we always access the hidden_states of its children, which are already initialized because we go from bottom to up.

Feel free to let me know if you have further questions.

[vrjmath]

Perfect, thanks!