|
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