From 1611fd5f65fc9d56a891dda024f7d407dd567eb8 Mon Sep 17 00:00:00 2001
From: Etienne Pot <epot@google.com>
Date: Thu, 27 Mar 2025 03:12:45 -0700
Subject: [PATCH] Remove pytype hack for the ContextStack typing annotations

PiperOrigin-RevId: 741084913
---
 etils/edc/stack.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/etils/edc/stack.py b/etils/edc/stack.py
index 7cb7766c..ab35c8d4 100644
--- a/etils/edc/stack.py
+++ b/etils/edc/stack.py
@@ -17,7 +17,6 @@
 from __future__ import annotations
 
 import dataclasses
-import typing
 from typing import Generic, TypeVar
 
 from etils.edc import context
@@ -54,12 +53,7 @@ def my_thread_safe_contextmanager():
   ```
   """
 
-  if typing.TYPE_CHECKING:
-    stack: list[_T] = dataclasses.field(default_factory=list)
-  else:
-    stack: context.ContextVar[list[_T]] = dataclasses.field(
-        default_factory=list
-    )
+  stack: context.ContextVar[list[_T]] = dataclasses.field(default_factory=list)
 
   def append(self, value: _T) -> None:
     """Append a value to the stack."""