16
16
import logging
17
17
import os
18
18
from types import TracebackType
19
- from typing import Any , Dict , List , Optional , Type , cast
19
+ from typing import Any , Optional , cast
20
20
from uuid import uuid4
21
21
22
22
import click
41
41
CONFIG_FILENAME = "config.yaml"
42
42
43
43
44
- def on_error (error : Exception , batch : List [ Dict [str , Any ]]) -> None :
44
+ def on_error (error : Exception , batch : list [ dict [str , Any ]]) -> None :
45
45
"""Custom error handler for Segment analytics.
46
46
47
47
Args:
@@ -88,7 +88,7 @@ def __enter__(self) -> "MLStacksAnalyticsContext":
88
88
89
89
def __exit__ (
90
90
self ,
91
- exc_type : Optional [Type [BaseException ]],
91
+ exc_type : Optional [type [BaseException ]],
92
92
exc_val : Optional [BaseException ],
93
93
exc_tb : Optional [TracebackType ],
94
94
) -> bool :
@@ -109,7 +109,7 @@ def __exit__(
109
109
def track (
110
110
self ,
111
111
event : AnalyticsEventsEnum ,
112
- properties : Optional [Dict [Any , Any ]] = None ,
112
+ properties : Optional [dict [Any , Any ]] = None ,
113
113
) -> Any :
114
114
"""Tracks event in Segment.
115
115
@@ -168,7 +168,7 @@ def set_analytics_user_id(user_id: str) -> None:
168
168
169
169
def track_event (
170
170
event : AnalyticsEventsEnum ,
171
- metadata : Optional [Dict [str , Any ]] = None ,
171
+ metadata : Optional [dict [str , Any ]] = None ,
172
172
) -> bool :
173
173
"""Track segment event if user opted-in.
174
174
@@ -195,7 +195,7 @@ class EventHandler:
195
195
def __init__ (
196
196
self ,
197
197
event : AnalyticsEventsEnum ,
198
- metadata : Optional [Dict [str , Any ]] = None ,
198
+ metadata : Optional [dict [str , Any ]] = None ,
199
199
):
200
200
"""Initialization of the context manager.
201
201
@@ -204,7 +204,7 @@ def __init__(
204
204
metadata: The metadata of the event.
205
205
"""
206
206
self .event : AnalyticsEventsEnum = event
207
- self .metadata : Dict [str , Any ] = metadata or {}
207
+ self .metadata : dict [str , Any ] = metadata or {}
208
208
209
209
def __enter__ (self ) -> "EventHandler" :
210
210
"""Enter function of the event handler.
0 commit comments