Water supply has been restored in certain areas of Johannesburg after a temporary shutdown.