ModuleNotFoundError: No module named 'agatha.ml.util.embedding_index'

[p123hx]

Hi,
When following the README to loading the Agatha model in
model = torch.load("<AGATHA_DATA_DIR>/hypothesis_predictor_512/model.pt")
, it prompts
ModuleNotFoundError: No module named 'agatha.ml.util.embedding_index'.
I looked up the directory agatha/ml/util/ and there is no file named embedding_index.
Could u tell how to solve it?
Thx

[p123hx]

It seems that in commit 8b4cd22 embedding_index is replaced by embedding_lookup.
Need to replace the file name accordingly.

[HarrySong123]

It seems that the pretrained model and the latest codes do not match. Replacing the file name of embedding_lookup is not enough since the repo also lacks some scripts that is asked by the pretrained model.

[amir-hassan25]

I am facing the same issue.