FIX: Ensure UITK Editor restores selection after domain reload (#1794)

Author: jamesmcgill

Packages/com.unity.inputsystem/CHANGELOG.md

@@ -43,6 +43,7 @@ however, it has to be formatted properly to pass verification tests.
 - Fix for BindingSyntax `WithInteraction()` which was incorrectly using processors.
 - Fix for UITK Input Action Editor binding 'Listen' button which wasn't working in the case for Control Type 'Any'.
 - Fixed issue of visual elements being null during editing project-wide actions in project settings which prompted console errors.
+- Fixed issue with UI Toolkit based Input Action Editor not restoring it's selected items after Domain Reload.
 
 
 ## [1.8.0-pre.1] - 2023-09-04

Packages/com.unity.inputsystem/InputSystem/Editor/UITKAssetEditor/InputActionsEditorWindow.cs

@@ -176,6 +176,7 @@ private void BuildUI()
         private void OnStateChanged(InputActionsEditorState newState)
         {
             DirtyInputActionsEditorWindow(newState);
+            m_State = newState;
 
             #if UNITY_INPUT_SYSTEM_INPUT_ACTIONS_EDITOR_AUTO_SAVE_ON_FOCUS_LOST
             // No action taken apart from setting dirty flag, auto-save triggered as part of having a dirty asset