[native] Add type params to setStateForContainer invocation in setRegisterState

Summary:
Following the Flow upgrade, I'm looking at replacing all usages of $Shape in our codebase. Context here.

When I swapped this particular $Shape into a Partial, it caused a Flow error. The Flow error was solved by adding type params to the setStateForContainer invocation.

Depends on D10083

Test Plan: Flow

Reviewers: michal

Reviewed By: michal

Subscribers: tomek

Differential Revision: https://phab.comm.dev/D10084