1616import logging
1717import os
1818from types import TracebackType
19- from typing import Any , Dict , List , Optional , Type , cast
19+ from typing import Any , Optional , cast
2020from uuid import uuid4
2121
2222import click
4141CONFIG_FILENAME = "config.yaml"
4242
4343
44- def on_error (error : Exception , batch : List [ Dict [str , Any ]]) -> None :
44+ def on_error (error : Exception , batch : list [ dict [str , Any ]]) -> None :
4545 """Custom error handler for Segment analytics.
4646
4747 Args:
@@ -88,7 +88,7 @@ def __enter__(self) -> "MLStacksAnalyticsContext":
8888
8989 def __exit__ (
9090 self ,
91- exc_type : Optional [Type [BaseException ]],
91+ exc_type : Optional [type [BaseException ]],
9292 exc_val : Optional [BaseException ],
9393 exc_tb : Optional [TracebackType ],
9494 ) -> bool :
@@ -109,7 +109,7 @@ def __exit__(
109109 def track (
110110 self ,
111111 event : AnalyticsEventsEnum ,
112- properties : Optional [Dict [Any , Any ]] = None ,
112+ properties : Optional [dict [Any , Any ]] = None ,
113113 ) -> Any :
114114 """Tracks event in Segment.
115115
@@ -168,7 +168,7 @@ def set_analytics_user_id(user_id: str) -> None:
168168
169169def track_event (
170170 event : AnalyticsEventsEnum ,
171- metadata : Optional [Dict [str , Any ]] = None ,
171+ metadata : Optional [dict [str , Any ]] = None ,
172172) -> bool :
173173 """Track segment event if user opted-in.
174174
@@ -195,7 +195,7 @@ class EventHandler:
195195 def __init__ (
196196 self ,
197197 event : AnalyticsEventsEnum ,
198- metadata : Optional [Dict [str , Any ]] = None ,
198+ metadata : Optional [dict [str , Any ]] = None ,
199199 ):
200200 """Initialization of the context manager.
201201
@@ -204,7 +204,7 @@ def __init__(
204204 metadata: The metadata of the event.
205205 """
206206 self .event : AnalyticsEventsEnum = event
207- self .metadata : Dict [str , Any ] = metadata or {}
207+ self .metadata : dict [str , Any ] = metadata or {}
208208
209209 def __enter__ (self ) -> "EventHandler" :
210210 """Enter function of the event handler.
0 commit comments