fix(dataGrid): focus loss on radio select (#6073)

* fix(dataGrid): focus loss on radio select

* fix: added optional ? on document

* chore: added conditional ? on document

packages/ibm-products/src/components/Datagrid/useSelectRows.tsx

@@ -102,7 +102,7 @@ const SelectRow = (datagridState) => {
     return () => window.removeEventListener('resize', updateSize);
   }, []);
 
-  const onSelectHandler = (event) => {
+  const onSelectHandler = async (event) => {
     event.stopPropagation(); // avoid triggering onRowClick
     if (radio) {
       toggleAllRowsSelected(false);
@@ -119,6 +119,10 @@ const SelectRow = (datagridState) => {
       getRowId,
       selectAll: null,
     });
+    // focus the radio / checkbox if lost
+    const activeElement = document?.activeElement?.id ?? '';
+    await undefined;
+    document?.getElementById(activeElement)?.focus();
   };
 
   const selectDisabled = isFetching || row.getRowProps().disabled;