search

IBM / regression-transformer

Regression Transformer (2023; Nature Machine Intelligence)
https://www.nature.com/articles/s42256-023-00639-z
MIT License
152 stars 23 forks source link

Missing BeamSearch class #4

Closed mpaszkow closed 2 years ago

mpaszkow commented 2 years ago

The BeamSearch class is not implemented in the terminator/search.py module, but it's imported in the terminator/evaluator.py file and used by the Evaluator class. Any attempt to import the Evaluator (e.g. in terminator/trainer.py) leads to an error:

ImportError: cannot import name 'BeamSearch' from 'terminator.search' (search.py)

jannisborn commented 2 years ago

Hi @mpaszkow, Thanks for the interest in our work. You're totally right, my bad. I will create a PR to fix this!