diff --git a/locust/dispatch.py b/locust/dispatch.py index 5542f8a117..5ea7adec61 100644 --- a/locust/dispatch.py +++ b/locust/dispatch.py @@ -90,13 +90,13 @@ def __init__(self, worker_nodes: list[WorkerNode], user_classes: list[type[User] assert len(user_classes) > 0 assert len(set(self._user_classes)) == len(self._user_classes) - self._target_user_count: int = None + self._target_user_count: int = 0 - self._spawn_rate: float = None + self._spawn_rate: float = 0.0 - self._user_count_per_dispatch_iteration: int = None + self._user_count_per_dispatch_iteration: int = 0 - self._wait_between_dispatch: float = None + self._wait_between_dispatch: float = 0.0 self._initial_users_on_workers = { worker_node.id: {user_class.__name__: 0 for user_class in self._user_classes} @@ -107,7 +107,8 @@ def __init__(self, worker_nodes: list[WorkerNode], user_classes: list[type[User] self._current_user_count = self.get_current_user_count() - self._dispatcher_generator: Generator[dict[str, dict[str, int]], None, None] = None + self._dispatcher_generator: Generator[dict[str, dict[str, int]], None, None] = None # type: ignore + # a generator is assigned (in new_dispatch()) to _dispatcher_generator before it's used self._user_generator = self._user_gen() @@ -131,7 +132,7 @@ def get_current_user_count(self) -> int: return sum(map(sum, map(dict.values, self._users_on_workers.values()))) @property - def dispatch_in_progress(self): + def dispatch_in_progress(self) -> bool: return self._dispatch_in_progress @property diff --git a/pyproject.toml b/pyproject.toml index 4227ac1f47..56841bacda 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -203,10 +203,6 @@ locust = ["locust"] ignore_missing_imports = true python_version = "3.9" -[[tool.mypy.overrides]] -module = ["locust.dispatch"] -strict_optional = false - [tool.pyright] exclude = ["locust/test"] reportOptionalMemberAccess = "none"