Most all the TMCC scale steam engines made 1998-2007 use Pittmans. Look up those engines and you will find a few flywheels still available. They will likely have magnets on them but will work. Pittmans have a 4mm shaft. I believe Canons have some smaller so you could use those after enlarging the bore.
Pete