a problem in ./pytorch/bts_test.py

[sigloome]

The class BtsModel is used in line 89 without importing

model = BtsModel(params=args)

Just add the following line to fix it
from bts import BtsModel

[cogaplex-bts]

@sigloome Thanks for the bug fixing. We will update the code.