AttributeError: 'KnowledgeGraph' object has no attribute 'entity2bucketid'

[BlackLucy]

pn.py L236 entity2bucketid = kg.entity2bucketid[e.tolist()]

There is an error 【AttributeError: 'KnowledgeGraph' object has no attribute 'entity2bucketid'】 in the 236th line of pn.py, how can I solve it?

[HernandezEduin]

Ensure you have the flag "--use_action_space_bucketing" active while running the command, this will calculate the entity2bucket in the KnowledgeGraph class.