Update to use python 3.12 syntax for generics

ArjanCodes · Feb 21, 2024 · d7f22df · d7f22df
1 parent 08d2253
commit d7f22df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/2024/tuesday_tips/generics/stack.py b/2024/tuesday_tips/generics/stack.py
@@ -1,8 +1,7 @@
-from typing import Optional, TypeVar, Generic
+from typing import Optional
 
-T = TypeVar('T') 
 
-class Stack(Generic[T]):
+class Stack[T]:
     def __init__(self) -> None:
         self._container: list[T] = []
 

diff --git a/2024/tuesday_tips/generics/vector_stack.py b/2024/tuesday_tips/generics/vector_stack.py
@@ -1,10 +1,7 @@
-from typing import TypeVar
 from stack import Stack
 
-T = TypeVar('T', int, float)
 
-
-class VectorStack(Stack[T]):
+class VectorStack[T: (int, float)](Stack[T]):
     def __getitem__(self, index: int) -> T:
         return self._container[index]