There is a train from Huangshan North to Wuyishan North it takes 1h23, Is it practical?

I want to spend a day at Huangshan and afterwards spend a day at Wuyishan. I read somewhere there is a train that takes about 10h from Wuyishan to Huangshan, but I don't see a return. rome2rio.com says it takes 3h from Wuyishan East to Wuyishan, is the train station really that far from Wuyishan or not good accecable? What alternatives are there? I inten to do a bamboo boat trip in Wuyishan.