Typically you would enter the amount from the mock federal return. However, since you both live in a community property state, your spouse is considered to have CA income (their share of the community income that you earned in CA). Since your spouse has CA income, CA law doesn't allow you to file separately if you file jointly for federal tax purposes, even though your spouse is a nonresident. So, in your case you'd enter the amount from the jointly filed federal return since you're required to file jointly in California.
@shreyasjoshi0906
