|
17 | 17 | Callable[..., Callable[..., Awaitable[Any]]], |
18 | 18 | _database_sync_to_async, |
19 | 19 | ) |
20 | | -WebsocketContext: Context[IdomWebsocket | None] = create_context(None) |
| 20 | +WebsocketContext: Context[Union[IdomWebsocket, None]] = create_context(None) |
21 | 21 | _REFETCH_CALLBACKS: DefaultDict[ |
22 | 22 | Callable[..., Any], set[Callable[[], None]] |
23 | 23 | ] = DefaultDict(set) |
@@ -45,10 +45,10 @@ def use_websocket() -> IdomWebsocket: |
45 | 45 |
|
46 | 46 |
|
47 | 47 | def use_query( |
48 | | - query: Callable[_Params, _Result | None], |
| 48 | + query: Callable[_Params, Union[_Result, None]], |
49 | 49 | *args: _Params.args, |
50 | 50 | **kwargs: _Params.kwargs, |
51 | | -) -> Query[_Result | None]: |
| 51 | +) -> Query[Union[_Result, None]]: |
52 | 52 | query_ref = use_ref(query) |
53 | 53 | if query_ref.current is not query: |
54 | 54 | raise ValueError(f"Query function changed from {query_ref.current} to {query}.") |
@@ -96,8 +96,8 @@ def execute_query() -> None: |
96 | 96 |
|
97 | 97 |
|
98 | 98 | def use_mutation( |
99 | | - mutate: Callable[_Params, bool | None], |
100 | | - refetch: Callable[..., Any] | Sequence[Callable[..., Any]], |
| 99 | + mutate: Callable[_Params, Union[bool, None]], |
| 100 | + refetch: Union[Callable[..., Any], Sequence[Callable[..., Any]]], |
101 | 101 | ) -> Mutation[_Params]: |
102 | 102 | loading, set_loading = use_state(False) |
103 | 103 | error, set_error = use_state(cast(Union[Exception, None], None)) |
|
0 commit comments