Review fixes

cmapi/integration_tests/utils.py

@@ -1,19 +1,14 @@
-from contextlib import contextmanager
 import os
+from collections.abc import Generator
+from contextlib import contextmanager
 
 
 @contextmanager
-def change_directory(new_dir):
+def change_directory(new_dir: str) -> Generator[None, None, None]:
     """
     Context manager for temporarily changing the current working directory.
 
-    Args:
-        new_dir: Directory to change to
-
-    Example:
-        with change_directory("/path/to/dir"):
-            # Code here runs with the working directory as "/path/to/dir"
-        # After the block, we return to the original directory
+    :param new_dir: Directory to change to
     """
     old_dir = os.getcwd()
     try: