Import from typing instead of typing_extensions, now that we are only…

… supporting 3.8+

locust/runners.py

@@ -29,15 +29,9 @@
     Any,
     cast,
     Callable,
+    TypedDict,
 )
 from uuid import uuid4
-
-# @TODO: typing.Protocol is in python >= 3.8
-try:
-    from typing import Protocol, TypedDict
-except ImportError:
-    from typing_extensions import Protocol, TypedDict  # type: ignore
-
 import gevent
 import greenlet
 import psutil
@@ -909,13 +903,13 @@ def check_stopped(self) -> None:
             not self.state == STATE_INIT
             and not self.state == STATE_STOPPED
             and (
-                    self.state == STATE_STOPPING
-                    and all(
-                        map(
-                            lambda x: x.state == STATE_INIT,
-                            self.clients.all,
-                        )
+                self.state == STATE_STOPPING
+                and all(
+                    map(
+                        lambda x: x.state == STATE_INIT,
+                        self.clients.all,
                     )
+                )
             )
             or all(
                 map(

locust/stats.py

@@ -28,14 +28,10 @@
     Callable,
     TypeVar,
     cast,
+    Protocol,
+    TypedDict,
 )
 
-# @TODO: typing.Protocol is in python >= 3.8
-try:
-    from typing import Protocol, TypedDict
-except ImportError:
-    from typing_extensions import Protocol, TypedDict  # type: ignore
-
 from types import FrameType
 
 from .exception import CatchResponseError

locust/user/task.py

@@ -13,14 +13,10 @@
     overload,
     Dict,
     Set,
+    Protocol,
+    final,
+    runtime_checkable,
 )
-
-# @TODO: typing.Protocol and typing.final is in python >= 3.8
-try:
-    from typing import Protocol, final, runtime_checkable
-except ImportError:
-    from typing_extensions import Protocol, final, runtime_checkable  # type: ignore
-
 import gevent
 from gevent import GreenletExit
 

locust/user/users.py

@@ -1,9 +1,8 @@
 from __future__ import annotations
-from typing import Callable, Dict, List, Optional
+from typing import Callable, Dict, List, Optional, final
 
 from gevent import GreenletExit, greenlet
 from gevent.pool import Group
-from typing_extensions import final
 from urllib3 import PoolManager
 
 from locust.clients import HttpSession