Add slots to dataclasses with Python 3.10+

aio-libs · Oct 3, 2024 · 946444b · 946444b
1 parent f289336
commit 946444b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aiohttp/helpers.py b/aiohttp/helpers.py
@@ -26,6 +26,7 @@
 from pathlib import Path
 from types import TracebackType
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
     ContextManager,
@@ -60,7 +61,9 @@
 else:
     import async_timeout
 
-if sys.version_info < (3, 10):
+if TYPE_CHECKING:
+    from dataclasses import dataclass as frozen_dataclass_decorator
+elif sys.version_info < (3, 10):
     frozen_dataclass_decorator = functools.partial(dataclasses.dataclass, frozen=True)
 else:
     frozen_dataclass_decorator = functools.partial(