change imports for pyi compat

reflex/components/moment/moment.py

@@ -1,7 +1,7 @@
 """Moment component for humanized date rendering."""
 
 import dataclasses
-import datetime
+from datetime import date, datetime, time, timedelta
 from typing import List, Optional, Union
 
 from reflex.components.component import NoSSRComponent
@@ -79,9 +79,7 @@ class Moment(NoSSRComponent):
     duration: Var[str]
 
     # The date to display (also work if passed as children).
-    date: Var[
-        Union[str, datetime.datetime, datetime.date, datetime.time, datetime.timedelta]
-    ]
+    date: Var[Union[str, datetime, date, time, timedelta]]
 
     # Shows the duration (elapsed time) between now and the provided datetime.
     duration_from_now: Var[bool]

reflex/components/moment/moment.pyi

@@ -4,7 +4,7 @@
 # This file was generated by `reflex/utils/pyi_generator.py`!
 # ------------------------------------------------------
 import dataclasses
-import datetime
+from datetime import date, datetime, time, timedelta
 from typing import Any, Dict, Optional, Union, overload
 
 from reflex.components.component import NoSSRComponent