1- from __future__ import annotations
2-
1+ import pathlib
32import time
43import weakref
54from dataclasses import dataclass
6- from typing import TYPE_CHECKING , Callable , ClassVar , Generator
5+ from typing import TYPE_CHECKING , Callable , ClassVar , Generator , Set , Union
76from unittest .mock import MagicMock , patch
87
98import anywidget ._descriptor
1514 ReprMimeBundle ,
1615)
1716from anywidget ._file_contents import FileContents
17+ from anywidget ._protocols import AnywidgetProtocol
1818from anywidget ._util import _WIDGET_MIME_TYPE
19+ from ipykernel .comm import Comm
1920from watchfiles import Change
2021
21- if TYPE_CHECKING :
22- import pathlib
23-
24- from anywidget ._protocols import AnywidgetProtocol
25- from ipykernel .comm import Comm
26-
2722
2823class MockComm (MagicMock ):
2924 # The only thing we need to do is to be able to relay messages back to
@@ -82,7 +77,7 @@ class Foo:
8277 _repr_mimebundle_ = MimeBundleDescriptor (autodetect_observer = False )
8378 value : int = val
8479
85- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
80+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
8681 return {"value" : self .value }
8782
8883 def __repr__ (self ) -> str :
@@ -122,7 +117,7 @@ def test_state_setter(mock_comm: MagicMock) -> None:
122117 class Foo :
123118 _repr_mimebundle_ = MimeBundleDescriptor (autodetect_observer = False )
124119
125- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
120+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
126121 return {}
127122
128123 def _set_anywidget_state (self , state ) -> None : # noqa: ANN001
@@ -142,7 +137,7 @@ def test_state_setter_binary(mock_comm: MagicMock) -> None:
142137 class Foo :
143138 _repr_mimebundle_ = MimeBundleDescriptor (autodetect_observer = False )
144139
145- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
140+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
146141 return {}
147142
148143 def _set_anywidget_state (self , state : dict ) -> None :
@@ -168,7 +163,7 @@ def test_comm_cleanup() -> None:
168163 class Foo :
169164 _repr_mimebundle_ = MimeBundleDescriptor (autodetect_observer = False )
170165
171- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
166+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
172167 return {}
173168
174169 foo = Foo ()
@@ -192,7 +187,7 @@ def test_detect_observer() -> None:
192187 class Foo :
193188 _repr_mimebundle_ = MimeBundleDescriptor ()
194189
195- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
190+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
196191 return {}
197192
198193 with pytest .warns (UserWarning , match = "Could not find a notifier" ):
@@ -208,7 +203,7 @@ class Foo:
208203 _repr_mimebundle_ = MimeBundleDescriptor (autodetect_observer = False )
209204 value : int = 1
210205
211- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
206+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
212207 return {"value" : self .value }
213208
214209 with pytest .warns (UserWarning , match = ".*is not weakrefable" ):
@@ -340,7 +335,7 @@ class Foo:
340335 _repr_mimebundle_ = MimeBundleDescriptor (_esm = esm , autodetect_observer = False )
341336 value : int = 1
342337
343- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
338+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
344339 return {"value" : self .value }
345340
346341 file_contents = Foo ._repr_mimebundle_ ._extra_state ["_esm" ]
@@ -388,7 +383,7 @@ class Foo:
388383 _repr_mimebundle_ = MimeBundleDescriptor (bar = bar , autodetect_observer = False )
389384 value : int = 1
390385
391- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
386+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
392387 return {"value" : self .value }
393388
394389 file_contents = Foo ._repr_mimebundle_ ._extra_state ["bar" ]
@@ -415,7 +410,7 @@ class Foo:
415410 autodetect_observer = False ,
416411 )
417412
418- def _get_anywidget_state (self , include : set [ str ] | None ): # noqa: ANN202, ARG002
413+ def _get_anywidget_state (self , include : Union [ Set [ str ], None ] ): # noqa: ANN202, ARG002
419414 return {}
420415
421416 foo = Foo ()
0 commit comments