Ensure we don't accidentally hide errors while importing in locust-cl…

…oud or locust-plugins
locustio · Aug 30, 2024 · 3b069d8 · 3b069d8
1 parent 8e98d84
commit 3b069d8
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/locust/main.py b/locust/main.py
@@ -39,12 +39,14 @@
 # import external plugins if  installed to allow for registering custom arguments etc
 try:
     import locust_plugins  # pyright: ignore[reportMissingImports]
-except ModuleNotFoundError:
-    pass
+except ModuleNotFoundError as e:
+    if e.msg != "No module named 'locust_plugins'":
+        raise
 try:
     import locust_cloud  # pyright: ignore[reportMissingImports]
-except ModuleNotFoundError:
-    pass
+except ModuleNotFoundError as e:
+    if e.msg != "No module named 'locust_cloud'":
+        raise
 
 version = locust.__version__