[Arr] group_by: throw when $key_fun returns a non-arraykey type

src/Psl/Arr/group_by.php

@@ -45,6 +45,8 @@ function group_by(iterable $values, callable $key_func): array
             continue;
         }
 
+        Psl\invariant(is_arraykey($key), 'Expected $key_func to return a value of type array-key, value of type (%s) returned.', gettype($key));
+        /** @psalm-var Tk $key */
         $result[$key] = $result[$key] ?? [];
         $result[$key][] = $value;
     }