Relying on something like train timings to enforce your work life balance isn’t a reliable strategy.

What if the trains started running every 5 minutes? Or what if you moved a little closer and you could walk?

If you don’t want to use willpower to stop working you could add some structured activity (after COVID) like a workout class after work. Or you could generally just make more deliberate plans for you after work time.

Or my personal favorite. Start a hobby you like more than work.