I understood it was this way too.
ATO suggests pro-rata is correct - there is a worked example at the bottom of this page https://www.ato.gov.au/General/Capital-gains-tax/In-detail/Real-estate/Treating-a-dwelling-as-your-main-residence-after-you-move-out/
