def simple_func(a: int, b: int) -> int:
        """
        A simple test function.

        Args:
          a: an int
          b: an int

        Returns:
          int: The sum of a and b.
        """

        def _internal(x: int) -> int:
            return x + a

        return _internal(b)

        A simple test function.

        Args:
          a: an int
          b: an int

        Returns:
          int: The sum of a and b.
        
ai/core/tests/core/oss/test_oss_client.py