Source code for ray.util.annotations

from typing import Optional
import inspect
import sys
import warnings
from functools import wraps


[docs]def PublicAPI(*args, **kwargs): """Annotation for documenting public APIs. Public APIs are classes and methods exposed to end users of Ray. If ``stability="alpha"``, the API can be used by advanced users who are tolerant to and expect breaking changes. If ``stability="beta"``, the API is still public and can be used by early users, but are subject to change. If ``stability="stable"``, the APIs will remain backwards compatible across minor Ray releases (e.g., Ray 1.4 -> 1.8). For a full definition of the stability levels, please refer to the :ref:`Ray API Stability definitions <api-stability>`. Args: stability: One of {"stable", "beta", "alpha"}. Examples: >>> from ray.util.annotations import PublicAPI >>> @PublicAPI ... def func(x): ... return x >>> @PublicAPI(stability="beta") ... def func(y): ... return y """ if len(args) == 1 and len(kwargs) == 0 and callable(args[0]): return PublicAPI(stability="stable")(args[0]) if "stability" in kwargs: stability = kwargs["stability"] assert stability in ["stable", "beta", "alpha"], stability elif kwargs: raise ValueError("Unknown kwargs: {}".format(kwargs.keys())) else: stability = "stable" def wrap(obj): if stability in ["alpha", "beta"]: message = ( f"**PublicAPI ({stability}):** This API is in {stability} " "and may change before becoming stable." ) _append_doc(obj, message=message) _mark_annotated(obj) return obj return wrap
[docs]def DeveloperAPI(*args, **kwargs): """Annotation for documenting developer APIs. Developer APIs are lower-level methods explicitly exposed to advanced Ray users and library developers. Their interfaces may change across minor Ray releases. Examples: >>> from ray.util.annotations import DeveloperAPI >>> @DeveloperAPI ... def func(x): ... return x """ if len(args) == 1 and len(kwargs) == 0 and callable(args[0]): return DeveloperAPI()(args[0]) def wrap(obj): _append_doc( obj, message="**DeveloperAPI:** This API may change across minor Ray releases.", ) _mark_annotated(obj) return obj return wrap
class RayDeprecationWarning(DeprecationWarning): """Specialized Deprecation Warning for fine grained filtering control""" pass # By default, print the first occurrence of matching warnings for # each module where the warning is issued (regardless of line number) if not sys.warnoptions: warnings.filterwarnings("module", category=RayDeprecationWarning)
[docs]def Deprecated(*args, **kwargs): """Annotation for documenting a deprecated API. Deprecated APIs may be removed in future releases of Ray. Args: message: a message to help users understand the reason for the deprecation, and provide a migration path. Examples: >>> from ray.util.annotations import Deprecated >>> @Deprecated ... def func(x): ... return x >>> @Deprecated(message="g() is deprecated because the API is error " ... "prone. Please call h() instead.") ... def g(y): ... return y """ if len(args) == 1 and len(kwargs) == 0 and callable(args[0]): return Deprecated()(args[0]) doc_message = ( "**DEPRECATED**: This API is deprecated and may be removed " "in future Ray releases." ) warning_message = ( "This API is deprecated and may be removed in future Ray releases. " "You could suppress this warning by setting env variable " 'PYTHONWARNINGS="ignore::DeprecationWarning"' ) warning = kwargs.pop("warning", False) if "message" in kwargs: doc_message = doc_message + "\n" + kwargs["message"] warning_message = warning_message + "\n" + kwargs["message"] del kwargs["message"] if kwargs: raise ValueError("Unknown kwargs: {}".format(kwargs.keys())) def inner(obj): _append_doc(obj, message=doc_message, directive="warning") _mark_annotated(obj) if not warning: return obj if inspect.isclass(obj): obj_init = obj.__init__ def patched_init(*args, **kwargs): warnings.warn(warning_message, RayDeprecationWarning, stacklevel=2) return obj_init(*args, **kwargs) obj.__init__ = patched_init return obj else: # class method or function. @wraps(obj) def wrapper(*args, **kwargs): warnings.warn(warning_message, RayDeprecationWarning, stacklevel=2) return obj(*args, **kwargs) return wrapper return inner
def _append_doc(obj, *, message: str, directive: Optional[str] = None) -> str: if not obj.__doc__: obj.__doc__ = "" obj.__doc__ = obj.__doc__.rstrip() indent = _get_indent(obj.__doc__) obj.__doc__ += "\n\n" if directive is not None: obj.__doc__ += f"{' ' * indent}.. {directive}::\n\n" message = message.replace("\n", "\n" + " " * (indent + 4)) obj.__doc__ += f"{' ' * (indent + 4)}{message}" else: message = message.replace("\n", "\n" + " " * (indent + 4)) obj.__doc__ += f"{' ' * indent}{message}" obj.__doc__ += f"\n{' ' * indent}" def _get_indent(docstring: str) -> int: """ Example: >>> def f(): ... '''Docstring summary.''' >>> f.__doc__ 'Docstring summary.' >>> _get_indent(f.__doc__) 0 >>> def g(foo): ... '''Docstring summary. ... ... Args: ... foo: Does bar. ... ''' >>> g.__doc__ 'Docstring summary.\\n\\n Args:\\n foo: Does bar.\\n ' >>> _get_indent(g.__doc__) 4 >>> class A: ... def h(): ... '''Docstring summary. ... ... Returns: ... None. ... ''' >>> A.h.__doc__ 'Docstring summary.\\n\\n Returns:\\n None.\\n ' >>> _get_indent(A.h.__doc__) 8 """ if not docstring: return 0 non_empty_lines = list(filter(bool, docstring.splitlines())) if len(non_empty_lines) == 1: # Docstring contains summary only. return 0 # The docstring summary isn't indented, so check the indentation of the second # non-empty line. return len(non_empty_lines[1]) - len(non_empty_lines[1].lstrip()) def _mark_annotated(obj) -> None: # Set magic token for check_api_annotations linter. if hasattr(obj, "__name__"): obj._annotated = obj.__name__ def _is_annotated(obj) -> bool: # Check the magic token exists and applies to this class (not a subclass). return hasattr(obj, "_annotated") and obj._annotated == obj.__name__