메인 콘텐츠로 건너뛰기

API 개요


Source

class DisplayNameFuncError


Source

class OpCallError


Source

class OpKwargs

op() 키워드 인수(keyword arguments)를 위한 TypedDict입니다.
Source

class Sentinel

Sentinel(package: ‘str’, path: ‘str’, name: ‘str’) Source

method __init__

__init__(package: 'str', path: 'str', name: 'str') → None

Source

class WeaveKwargs


Source

function as_op

as_op(fn: 'Callable[P, R]') → Op[P, R]
@weave.op 데코레이터가 적용된 함수가 주어지면, 해당 Op를 반환합니다. @weave.op 데코레이터가 적용된 함수는 이미 Op의 인스턴스이므로, 이 함수는 런타임에 no-op으로 동작해야 합니다. 하지만 타입 안전한 방식으로 OpDef 속성에 엑세스해야 하는 경우, 타입 체커를 만족시키기 위해 이 함수를 사용할 수 있습니다. Args:
  • fn: weave.op 데코레이터가 적용된 함수. Returns: 함수의 Op.

Source

function call

call(
    op: 'Op',
    *args: 'Any',
    __weave: 'WeaveKwargs | None' = None,
    __should_raise: 'bool' = False,
    __require_explicit_finish: 'bool' = False,
    **kwargs: 'Any'
) → tuple[Any, Call] | Coroutine[Any, Any, tuple[Any, Call]]
op를 실행하고 결과(result)와 실행을 나타내는 Call을 모두 반환합니다. 이 함수는 예외를 발생시키지 않습니다. 모든 에러는 Call 오브젝트에 캡처됩니다. 이 메소드는 @weave.op 데코레이터가 적용된 모든 함수에 자동으로 바인딩되어 다음과 같이 사용할 수 있습니다:
@weave.op
def add(a: int, b: int) -> int:
     return a + b

result, call = add.call(1, 2)

Source

function calls

calls(op: 'Op') → CallsIter
이 op에 대한 모든 호출에 대한 이터레이터(iterator)를 가져옵니다. 이 메소드는 @weave.op 데코레이터가 적용된 모든 함수에 자동으로 바인딩되어 다음과 같이 사용할 수 있습니다:
@weave.op
def add(a: int, b: int) -> int:
     return a + b

calls = add.calls()
for call in calls:
     print(call)

Source

function get_captured_code

get_captured_code(op: 'Op') → str
op의 캡처된 코드를 가져옵니다. 이 기능은 ref로부터 op를 다시 가져올 때만 작동합니다. 패턴은 다음과 같습니다: ref = weave.publish(func) op = ref.get() captured_code = op.get_captured_code()
Source

function is_op

is_op(obj: 'Any') → TypeIs[Op]
오브젝트가 Op인지 확인합니다.
Source

function is_placeholder_call

is_placeholder_call(call: 'Call') → TypeIs[NoOpCall]

Source

function is_tracing_setting_disabled

is_tracing_setting_disabled() → bool

Source

function maybe_bind_method

maybe_bind_method(func: 'Callable', self: 'Any' = None) → Callable | MethodType
함수를 임의의 오브젝트에 바인딩합니다 (클래스가 아니더라도 가능합니다). self가 None인 경우, 함수를 있는 그대로 반환합니다.
Source

function maybe_unbind_method

maybe_unbind_method(oplike: 'Op | MethodType | partial') → Op
Op와 유사한 메소드 또는 partial을 일반 Op 함수로 바인딩 해제합니다. 수행 내용:
  • 메소드의 경우, 설정된 self 파라미터를 제거합니다.
  • partial의 경우, 사전 설정된 모든 파라미터를 제거합니다.

Source

function op

op(
    func: 'Callable[P, R] | None' = None,
    name: 'str | None' = None,
    call_display_name: 'str | CallDisplayNameFunc | None' = None,
    postprocess_inputs: 'PostprocessInputsFunc | None' = None,
    postprocess_output: 'PostprocessOutputFunc | None' = None,
    tracing_sample_rate: 'float' = 1.0,
    enable_code_capture: 'bool' = True,
    accumulator: 'Callable[[Any | None, Any], Any] | None' = None,
    kind: 'OpKind | None' = None,
    color: 'OpColor | None' = None
) → Callable[[Callable[P, R]], Op[P, R]] | Op[P, R]
함수나 메소드를 weave op로 변환하는 데코레이터입니다. 동기 및 비동기 방식 모두에서 작동합니다. 이터레이터 함수를 자동으로 감지하고 적절한 행동을 적용합니다.
Source

function placeholder_call

placeholder_call() → Call

Source

function setup_dunder_weave_dict

setup_dunder_weave_dict(op: 'Op', d: 'WeaveKwargs | None' = None) → WeaveKwargs
WeaveKwargs를 op에 전달하는 데 사용되는 __weave 딕셔너리를 설정합니다. Args:
  • d: 업데이트할 선택 사항인 기존 WeaveKwargs 딕셔너리.
  • op: kind와 color를 추출할 Op. Returns: attributes, display_name 및 선택적으로 kind/color가 설정된 WeaveKwargs 딕셔너리.

Source

function should_skip_tracing_for_op

should_skip_tracing_for_op(op: 'Op') → bool