o $i @s ddlZddlZddlmZddlmZmZmZmZddlm Z e e Z dZ dedejfdd Zdded edeefd d ZdeedeedfdeeefddfddZdeedeedfdeeefdeefddZdeedeeeeeefffddZdS)N) Parameter)AnyDictListTuple) is_cythons __RAY_DUMMY__funcreturncshtr/gd}tfdd|Dr(}dd|D] }t|t||qntdtS)aBGet signature parameters. Support Cython functions by grabbing relevant attributes from the Cython function and attaching to a no-op function. This is somewhat brittle, since inspect may change, but given that inspect is written to a PEP, we hope it is relatively stable. Future versions of Python may allow overloading the inspect 'isfunction' and 'ismethod' functions / create ABC for Python functions. Until then, it appears that Cython won't do anything about compatability with the inspect module. Args: func: The function whose signature should be checked. Returns: A function signature object, which includes the names of the keyword arguments as well as their default values. Raises: TypeError: A type error if the signature is not supported )__code____annotations__ __defaults____kwdefaults__c3s|]}t|VqdSN)hasattr).0attrrR/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/_common/signature.py ,sz get_signature..cSsdSrrrrrrr/szget_signature..funcz( is not a Python function we can process)rallsetattrgetattr TypeErrorinspect signature)rattrs original_funcrrrr get_signatures rF ignore_firstcCsDtt|j}|r t|dkrtd|jd|dd}|S)a\Extract the function signature from the function. Args: func: The function whose signature should be extracted. ignore_first: True if the first argument should be ignored. This should be used when func is a method of a class. Returns: List of Parameter objects representing the function signature. rz5Methods must take a 'self' argument, but the method 'z' does not have one.N)listr parametersvalueslen ValueError__name__)rrsignature_parametersrrrextract_signature:s   r(r'args.kwargsc CsJtj|d}z |j|i|WdSty$}ztt|dd}~ww)aValidates the arguments against the signature. Args: signature_parameters: The list of Parameter objects representing the function signature, obtained from `extract_signature`. args: The positional arguments passed into the function. kwargs: The keyword arguments passed into the function. Raises: TypeError: Raised if arguments do not fit in the function signature. )r"N)r Signaturebindrstr)r'r)r*reconstructed_signatureexcrrr validate_argsRs r0cCsHt|||g}|D]}|t|g7}q |D] \}}|||g7}q|S)a Validates the arguments against the signature and flattens them. The flat list representation is a serializable format for arguments. Since the flatbuffer representation of function arguments is a list, we combine both keyword arguments and positional arguments. We represent this with two entries per argument value - [DUMMY_TYPE, x] for positional arguments and [KEY, VALUE] for keyword arguments. See the below example. See `recover_args` for logic restoring the flat list back to args/kwargs. Args: signature_parameters: The list of Parameter objects representing the function signature, obtained from `extract_signature`. args: The positional arguments passed into the function. kwargs: The keyword arguments passed into the function. Returns: List of args and kwargs. Non-keyword arguments are prefixed by internal enum DUMMY_TYPE. Raises: TypeError: Raised if arguments do not fit in the function signature. )r0 DUMMY_TYPEitems)r'r)r* list_argsargkeywordrrr flatten_argshs r6flattened_argscCspt|ddks Jdg}i}tdt|dD]}||||d}}|tkr/||q|||<q||fS)aDRecreates `args` and `kwargs` from the flattened arg list. Args: flattened_args: List of args and kwargs. This should be the output of `flatten_args`. Returns: args: The non-keyword arguments passed into the function. kwargs: The keyword arguments passed into the function. rzAFlattened arguments need to be even-numbered. See `flatten_args`.r )r$ranger1append)r7r)r* name_indexnamer4rrr recover_argss   r=)F)rloggingrtypingrrrrray._private.inspect_utilr getLoggerr&loggerr1r+rboolr(r-r0r6r=rrrrs8   (      .$