fix type error for GenericAlias

Textualize · Sep 27, 2023 · 6698bbb · 6698bbb
1 parent 584f3fc
commit 6698bbb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/tree/test_tree_availability.py b/tests/tree/test_tree_availability.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any
 
 from textual import on
@@ -33,19 +35,19 @@ def record(
         )
 
     @on(Tree.NodeSelected)
-    def node_selected(self, event: Tree.NodeSelected) -> None:
+    def node_selected(self, event: Tree.NodeSelected[None]) -> None:
         self.record(event)
 
     @on(Tree.NodeExpanded)
-    def node_expanded(self, event: Tree.NodeExpanded) -> None:
+    def node_expanded(self, event: Tree.NodeExpanded[None]) -> None:
         self.record(event)
 
     @on(Tree.NodeCollapsed)
-    def node_collapsed(self, event: Tree.NodeCollapsed) -> None:
+    def node_collapsed(self, event: Tree.NodeCollapsed[None]) -> None:
         self.record(event)
 
     @on(Tree.NodeHighlighted)
-    def node_highlighted(self, event: Tree.NodeHighlighted) -> None:
+    def node_highlighted(self, event: Tree.NodeHighlighted[None]) -> None:
         self.record(event)