Hi @xzhaoma
The modules you are trying to import are only present in the beta version of Flatland. You have to clone the git repository and build it from source.
Hope this helps.
Best regards,
Erik
PS: If you are using the new beta repository you can also try out sparse_rail_generator and sparse_schedule_generator to be ready for Round 2 of the challenge.