React - Add attribute to componente inside prop

example.tsx

/* trigger is the prop that accepts a ReactNode */

  const addAttr = useMemo(() => {
    const InputReactObject = Children.only(trigger);
    const clonedChild = cloneElement(inputReactObject, {
      className: 'input-element test',
      'data-attr': 'lorem' 
    });

    return <InputReactObject />;
  }, [trigger]);