11"""Base concept drift module."""
22
33import abc
4- from typing import Any , Dict , List , Optional , Union
4+ from typing import Any , Optional , Union
55
66from frouros .callbacks import HistoryConceptDrift
77from frouros .callbacks .streaming .base import BaseCallbackStreaming
@@ -66,7 +66,7 @@ def __init__(
6666 self ,
6767 config : Optional [BaseConceptDriftConfig ] = None ,
6868 callbacks : Optional [
69- Union [BaseCallbackStreaming , List [BaseCallbackStreaming ]]
69+ Union [BaseCallbackStreaming , list [BaseCallbackStreaming ]]
7070 ] = None ,
7171 ) -> None :
7272 """Init method.
@@ -75,7 +75,7 @@ def __init__(
7575 :type config: Optional[BaseConceptDriftConfig]
7676 :param callbacks: callbacks
7777 :type callbacks: Optional[Union[BaseCallbackStreaming,
78- List [BaseCallbackStreaming]]]
78+ list [BaseCallbackStreaming]]]
7979 """
8080 check_callbacks (
8181 callbacks = callbacks ,
@@ -99,20 +99,20 @@ def _set_additional_vars_callback(self) -> None:
9999 )
100100
101101 @property
102- def additional_vars (self ) -> Optional [Dict [str , Any ]]:
102+ def additional_vars (self ) -> Optional [dict [str , Any ]]:
103103 """Additional variables property.
104104
105105 :return: additional variables
106- :rtype: Optional[Dict [str, Any]]
106+ :rtype: Optional[dict [str, Any]]
107107 """
108108 return self ._additional_vars
109109
110110 @additional_vars .setter
111- def additional_vars (self , value : Optional [Dict [str , Any ]]) -> None :
111+ def additional_vars (self , value : Optional [dict [str , Any ]]) -> None :
112112 """Additional variables setter.
113113
114114 :param value: value to be set
115- :type value: Optional[Dict [str, Any]]
115+ :type value: Optional[dict [str, Any]]
116116 """
117117 self ._additional_vars = value if value is not None else {}
118118
@@ -171,21 +171,21 @@ def reset(self) -> None:
171171 callback .reset ()
172172
173173 @property
174- def status (self ) -> Dict [str , bool ]:
174+ def status (self ) -> dict [str , bool ]:
175175 """Status property.
176176
177177 :return: status dict
178- :rtype: Dict [str, bool]
178+ :rtype: dict [str, bool]
179179 """
180180 return {"drift" : self .drift }
181181
182- def update (self , value : Union [int , float ], ** kwargs ) -> Dict [str , Any ]:
182+ def update (self , value : Union [int , float ], ** kwargs ) -> dict [str , Any ]:
183183 """Update method.
184184
185185 :param value: value to update detector
186186 :type value: Union[int, float]
187187 :return: callbacks logs
188- :rtype: Dict [str, Any]]
188+ :rtype: dict [str, Any]]
189189 """
190190 for callback in self .callbacks : # type: ignore
191191 callback .on_update_start ( # type: ignore
@@ -200,7 +200,7 @@ def update(self, value: Union[int, float], **kwargs) -> Dict[str, Any]:
200200 callbacks_logs = self ._get_callbacks_logs ()
201201 return callbacks_logs
202202
203- def _get_callbacks_logs (self ) -> Dict [str , Any ]:
203+ def _get_callbacks_logs (self ) -> dict [str , Any ]:
204204 logs = {
205205 callback .name : callback .logs for callback in self .callbacks # type: ignore
206206 }
0 commit comments