Add imports of other modules to stubs

src/stub_generator.py

@@ -7,6 +7,8 @@
 
 from .code_generator import CodeGenerator
 
+MODULE_NAMES = ["carla", "command"]
+
 
 class CarlaStubGenerator:
     module_name: str
@@ -27,6 +29,9 @@ def __init__(self, file):
         self._generator.add_line("from typing import Any")
         self._generator.add_line("import numpy as np")
         self._generator.add_line()
+        for module_name in MODULE_NAMES:
+            if self.module_name != module_name:
+                self._generator.add_line(f"import {module_name}")
         self._generator.add_line()
 
     def save_to_file(self, directory: Path):
@@ -121,7 +126,9 @@ def _generate_methods(self, _class):
                 warning = method.get("warning", None)
                 if warning:
                     self._generator.add_line()
-                    self._generator.add_line(f"**warning**: {self._parse_docs(warning)}")
+                    self._generator.add_line(
+                        f"**warning**: {self._parse_docs(warning)}"
+                    )
 
                 for param in param_list:
                     param_name = param.get("param_name", "")

stubs/0.9.13/__init__.pyi

@@ -3,6 +3,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class Actor:
 	"""
@@ -807,6 +808,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class ActorAttributeType:
 	"""
@@ -1227,6 +1229,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class Client:
 	"""
@@ -1731,6 +1734,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class VehicleControl:
 	"""
@@ -2154,6 +2158,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class Vector2D:
 	"""
@@ -2846,6 +2851,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class LightGroup:
 	"""
@@ -3177,6 +3183,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class LaneType:
 	"""
@@ -3896,6 +3903,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class Osm2Odr:
 	"""
@@ -3971,6 +3979,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class Sensor(Actor):
 	"""
@@ -4169,6 +4178,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class SensorData:
 	"""
@@ -5056,6 +5066,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class WorldSnapshot:
 	"""
@@ -5174,6 +5185,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class WeatherParameters:
 	"""
@@ -5287,6 +5299,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import command
 
 class Timestamp:
 	"""

stubs/0.9.13/command.pyi

@@ -3,6 +3,7 @@ from __future__ import annotations
 from typing import Any
 import numpy as np
 
+import carla
 
 class Response:
 	"""