changes in tICA_train.py to match library changes