replace exempt_from_compat_check for/in favor of internal_ap

Signed-off-by: Pablo Garay <[email protected]>

Author: pablo-garay

megatron/core/utils.py

@@ -2159,31 +2159,6 @@ async def wrapper(*args, **kwargs):
 # ============================================================================
 
 
-def exempt_from_compat_check(func: Callable) -> Callable:
-    """
-    Mark a function as exempt from backward compatibility checks.
-    
-    Use this decorator for:
-    - Internal APIs not intended for external use
-    - Experimental features that may change
-    - Functions explicitly documented as unstable
-    
-    Args:
-        func: The function to mark as exempt
-    
-    Returns:
-        The original function with an exemption marker
-    
-    Example:
-        @exempt_from_compat_check
-        def experimental_api():
-            '''This API may change without notice'''
-            pass
-    """
-    func._exempt_from_compat_check = True
-    return func
-
-
 def deprecated(
     version: str,
     removal_version: Optional[str] = None,
@@ -2254,23 +2229,32 @@ def wrapper(*args, **kwargs):
 
 def internal_api(func: Callable) -> Callable:
     """
-    Mark a function as internal API (not for external use).
+    Mark a function or class as internal API (not for external use).
+    
+    Use this decorator for:
+    - Internal APIs not intended for public consumption
+    - Experimental features that may change without notice
+    - Implementation details that are not part of the stable API
     
-    This is semantically similar to exempt_from_compat_check but
-    more explicitly communicates that the function is internal.
+    Objects marked with this decorator will be exempt from backward
+    compatibility checks.
     
     Args:
-        func: The function to mark as internal
+        func: The function or class to mark as internal
     
     Returns:
-        The original function with an internal API marker
+        The original function/class with an internal API marker
     
     Example:
         @internal_api
         def _internal_helper():
             '''For internal use only'''
             pass
+        
+        @internal_api
+        class ExperimentalFeature:
+            '''This API may change without notice'''
+            pass
     """
     func._internal_api = True
-    func._exempt_from_compat_check = True  # Also exempt from checks
     return func

scripts/check_api_backwards_compatibility.py

@@ -3,7 +3,7 @@
 Megatron Core API Compatibility Checker
 
 Simple checker using Griffe to find breaking changes between two versions.
-Objects decorated with @exempt_from_compat_check are excluded from checks.
+Objects decorated with @internal_api or @deprecated are excluded from checks.
 
 Usage:
     python scripts/check_api_backwards_compatibility.py --baseline core_v0.14.0
@@ -43,7 +43,7 @@
 
 
 # Decorators that exempt objects from compatibility checks
-EXEMPT_DECORATORS = ['exempt_from_compat_check', 'deprecated', 'internal_api']
+EXEMPT_DECORATORS = ['internal_api', 'deprecated']
 
 
 def has_exempt_decorator(obj: Object) -> bool: