Update src/instamatic/microscope/components/stage.py

Co-authored-by: Stef Smeets <stefsmeets@users.noreply.github.com>

Author: Baharis

src/instamatic/microscope/components/stage.py

@@ -90,7 +90,7 @@ def set_a_with_speed(self, a: float, speed: int, wait: bool = False):
             self.wait()
 
     @contextmanager
-    def rotation_speed(self, speed: Union[float, int]) -> Generator[None, None, None]:
+    def rotation_speed(self, speed: Number) -> Generator[None, None, None]:
         """Context manager that sets the rotation speed for the duration of the
         `with` statement (JEOL, Tecnai only).