hebra 0.1.0__py3-none-macosx_11_0_arm64.whl

hebra/__init__.py

@@ -0,0 +1,51 @@
+"""hebra: SwanLab风格的实验追踪SDK（demo版本）。
+
+使用方式：
+    import hebra
+
+    # 1. 初始化（启动Go服务进程）
+    run = hebra.init(project="my-project", logdir="./logs")
+
+    # 2. 记录数据
+    hebra.log({"loss": 0.1, "acc": 0.9})
+    # 或通过run对象
+    run.log({"loss": 0.1})
+
+    # 3. 可选：手动结束（不调用会自动结束）
+    hebra.finish()
+"""
+
+from __future__ import annotations
+
+import sys
+from pathlib import Path
+
+# ============================================================================
+# Proto 路径配置（必须在导入模块前完成）
+# ============================================================================
+
+PKG_ROOT = Path(__file__).resolve().parent
+PKG_PROTO_ROOT = PKG_ROOT / "proto"
+
+if str(PKG_PROTO_ROOT) not in sys.path and PKG_PROTO_ROOT.exists():
+    sys.path.append(str(PKG_PROTO_ROOT))
+
+# ============================================================================
+# 导入核心 API
+# ============================================================================
+
+from .run import HebraRun, HebraRunState, get_run
+from .sdk import init, log, finish
+
+# ============================================================================
+# 导出
+# ============================================================================
+
+__all__ = [
+    "init",
+    "log",
+    "finish",
+    "get_run",
+    "HebraRun",
+    "HebraRunState",
+]

hebra/bin/.gitkeep

@@ -0,0 +1 @@
+

hebra/lib/service_connection.py

@@ -0,0 +1,137 @@
+from __future__ import annotations
+
+import os
+import shutil
+import subprocess
+import tempfile
+import time
+from pathlib import Path
+from typing import Optional
+
+import grpc
+
+# Proto imports
+from common.v1 import common_pb2
+from core.collector.v1 import (
+    collector_service_pb2,
+    collector_service_pb2_grpc,
+)
+from core.lifecycle.v1 import (
+    lifecycle_service_pb2,
+    lifecycle_service_pb2_grpc,
+)
+
+
+class ServiceConnection:
+    """管理与Go服务的连接（内部使用）"""
+
+    def __init__(self, db_path: str):
+        self._proc: Optional[subprocess.Popen] = None
+        self._channel: Optional[grpc.Channel] = None
+        self._workdir: Optional[Path] = None
+        self._db_path = db_path
+
+    def start(self) -> None:
+        """启动Go服务进程"""
+        if self._proc is not None:
+            return
+
+        server_bin = find_server_bin()
+        self._workdir = Path(tempfile.mkdtemp(prefix="hebra-"))
+        port_file = self._workdir / "port.txt"
+
+        self._proc = subprocess.Popen(
+            [
+                str(server_bin),
+                "--pid",
+                str(os.getpid()),
+                "--port-file",
+                str(port_file),
+                "--db-path",
+                self._db_path,
+            ],
+            cwd=self._workdir,
+        )
+
+        addr_line = wait_for_address(port_file)
+        kind, target = parse_address(addr_line)
+        self._channel = make_channel(kind, target)
+
+    def upload(
+        self, data: common_pb2.KeyValueList
+    ) -> collector_service_pb2.CollectorUploadResponse:
+        """上传数据到Go服务"""
+        if self._channel is None:
+            raise RuntimeError("服务未启动")
+        stub = collector_service_pb2_grpc.CollectorStub(self._channel)
+        req = collector_service_pb2.CollectorUploadRequest(data=data)
+        return stub.Upload(req, timeout=5)
+
+    def shutdown(self, exit_code: int = 0, timeout: float = 5.0) -> None:
+        """关闭Go服务"""
+        if self._channel is not None and self._proc is not None:
+            try:
+                stub = lifecycle_service_pb2_grpc.LifecycleStub(self._channel)
+                req = lifecycle_service_pb2.ShutdownRequest(exit_code=exit_code)
+                stub.Shutdown(req, timeout=2)
+            except grpc.RpcError:
+                pass
+            finally:
+                self._channel.close()
+                self._channel = None
+
+        if self._proc is not None:
+            try:
+                self._proc.wait(timeout=timeout)
+            except subprocess.TimeoutExpired:
+                self._proc.kill()
+                self._proc.wait()
+            self._proc = None
+
+        if self._workdir is not None:
+            shutil.rmtree(self._workdir, ignore_errors=True)
+            self._workdir = None
+
+
+# 内部辅助函数
+
+
+def find_server_bin() -> Path:
+    """查找Go服务二进制文件"""
+    from .. import PKG_ROOT
+
+    exe_name = "core.exe" if os.name == "nt" else "core"
+    bin_path = PKG_ROOT / "bin" / exe_name
+    if bin_path.exists() and bin_path.is_file():
+        return bin_path
+    raise FileNotFoundError(f"服务二进制文件未找到: {bin_path}")
+
+
+def wait_for_address(path: Path, timeout: float = 10.0) -> str:
+    """等待端口文件生成并读取地址"""
+    deadline = time.time() + timeout
+    while time.time() < deadline:
+        if path.exists():
+            text = path.read_text().strip()
+            if text:
+                return text.splitlines()[0].strip()
+        time.sleep(0.1)
+    raise TimeoutError(f"端口文件未在{timeout}秒内生成")
+
+
+def parse_address(line: str) -> tuple:
+    """解析地址格式"""
+    if line.startswith("unix://"):
+        return "unix", line[len("unix://") :]
+    if line.startswith("tcp://"):
+        return "tcp", line[len("tcp://") :]
+    raise ValueError(f"未知地址格式: {line}")
+
+
+def make_channel(kind: str, target: str) -> grpc.Channel:
+    """创建gRPC通道"""
+    if kind == "unix":
+        return grpc.insecure_channel(f"unix://{target}")
+    if target.startswith(":"):
+        target = f"localhost{target}"
+    return grpc.insecure_channel(target)

hebra/lib/utils.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from typing import Any, Mapping, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ..run import HebraRun
+
+# Proto imports (需要在包初始化时设置好 sys.path)
+from common.v1 import common_pb2
+
+
+def should_call_before_init(error_message: str):
+    """装饰器：限制必须在init之前调用"""
+    from ..run import HebraRun
+
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            if HebraRun.is_started():
+                raise RuntimeError(error_message)
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator
+
+
+def should_call_after_init(error_message: str):
+    """装饰器：限制必须在init之后调用"""
+    from ..run import HebraRun
+
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            if not HebraRun.is_started():
+                raise RuntimeError(error_message)
+            return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator
+
+
+def to_any(value: Any) -> common_pb2.AnyValue:
+    """将Python值转换为AnyValue"""
+    if isinstance(value, bool):
+        return common_pb2.AnyValue(bool_value=value)
+    if isinstance(value, int):
+        return common_pb2.AnyValue(int_value=value)
+    if isinstance(value, float):
+        return common_pb2.AnyValue(double_value=value)
+    if isinstance(value, str):
+        return common_pb2.AnyValue(string_value=value)
+    if isinstance(value, (list, tuple)):
+        arr = common_pb2.ArrayValue(values=[to_any(v) for v in value])
+        return common_pb2.AnyValue(array_value=arr)
+    # 其他类型转为字符串
+    return common_pb2.AnyValue(string_value=str(value))
+
+
+def kvs_from_mapping(kvs: Mapping[str, Any]) -> common_pb2.KeyValueList:
+    """将字典转换为KeyValueList"""
+    kv_list = common_pb2.KeyValueList()
+    for k, v in kvs.items():
+        if not k:
+            continue
+        kv_list.values.append(common_pb2.KeyValue(key=str(k), value=to_any(v)))
+    return kv_list

hebra/proto/common/v1/common.proto

@@ -0,0 +1,50 @@
+syntax = "proto3";
+
+package swanlab.proto.common.v1;
+
+option go_package = "core/pkg/pb";
+
+
+// AnyValue is used to represent any type of attribute value. AnyValue may contain a
+// primitive value such as a string or integer or it may contain an arbitrary nested
+// object containing arrays, key-value lists and primitives.
+message AnyValue {
+  // The value is one of the listed fields. It is valid for all values to be unspecified
+  // in which case this AnyValue is considered to be "empty".
+  oneof value {
+    string string_value = 1;
+    bool bool_value = 2;
+    int64 int_value = 3;
+    double double_value = 4;
+    ArrayValue array_value = 5;
+    KeyValueList kvlist_value = 6;
+    bytes bytes_value = 7;
+  }
+}
+
+// ArrayValue is a list of AnyValue messages. We need ArrayValue as a message
+// since oneof in AnyValue does not allow repeated fields.
+message ArrayValue {
+  // Array of values. The array may be empty (contain 0 elements).
+  repeated AnyValue values = 1;
+}
+
+// KeyValueList is a list of KeyValue messages. We need KeyValueList as a message
+// since `oneof` in AnyValue does not allow repeated fields. Everywhere else where we need
+// a list of KeyValue messages (e.g. in Span) we use `repeated KeyValue` directly to
+// avoid unnecessary extra wrapping (which slows down the protocol). The 2 approaches
+// are semantically equivalent.
+message KeyValueList {
+  // A collection of key/value pairs of key-value pairs. The list may be empty (may
+  // contain 0 elements).
+  // The keys MUST be unique (it is not allowed to have more than one
+  // value with the same key).
+  repeated KeyValue values = 1;
+}
+
+// KeyValue is a key-value pair that is used to store Span attributes, Link
+// attributes, etc.
+message KeyValue {
+  string key = 1;
+  AnyValue value = 2;
+}

hebra/proto/common/v1/common_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: common/v1/common.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'common/v1/common.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x16\x63ommon/v1/common.proto\x12\x17swanlab.proto.common.v1\"\x80\x02\n\x08\x41nyValue\x12\x16\n\x0cstring_value\x18\x01 \x01(\tH\x00\x12\x14\n\nbool_value\x18\x02 \x01(\x08H\x00\x12\x13\n\tint_value\x18\x03 \x01(\x03H\x00\x12\x16\n\x0c\x64ouble_value\x18\x04 \x01(\x01H\x00\x12:\n\x0b\x61rray_value\x18\x05 \x01(\x0b\x32#.swanlab.proto.common.v1.ArrayValueH\x00\x12=\n\x0ckvlist_value\x18\x06 \x01(\x0b\x32%.swanlab.proto.common.v1.KeyValueListH\x00\x12\x15\n\x0b\x62ytes_value\x18\x07 \x01(\x0cH\x00\x42\x07\n\x05value\"?\n\nArrayValue\x12\x31\n\x06values\x18\x01 \x03(\x0b\x32!.swanlab.proto.common.v1.AnyValue\"A\n\x0cKeyValueList\x12\x31\n\x06values\x18\x01 \x03(\x0b\x32!.swanlab.proto.common.v1.KeyValue\"I\n\x08KeyValue\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32!.swanlab.proto.common.v1.AnyValueB\rZ\x0b\x63ore/pkg/pbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'common.v1.common_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\013core/pkg/pb'
+  _globals['_ANYVALUE']._serialized_start=52
+  _globals['_ANYVALUE']._serialized_end=308
+  _globals['_ARRAYVALUE']._serialized_start=310
+  _globals['_ARRAYVALUE']._serialized_end=373
+  _globals['_KEYVALUELIST']._serialized_start=375
+  _globals['_KEYVALUELIST']._serialized_end=440
+  _globals['_KEYVALUE']._serialized_start=442
+  _globals['_KEYVALUE']._serialized_end=515
+# @@protoc_insertion_point(module_scope)

hebra/proto/common/v1/common_pb2.pyi

@@ -0,0 +1,45 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class AnyValue(_message.Message):
+    __slots__ = ("string_value", "bool_value", "int_value", "double_value", "array_value", "kvlist_value", "bytes_value")
+    STRING_VALUE_FIELD_NUMBER: _ClassVar[int]
+    BOOL_VALUE_FIELD_NUMBER: _ClassVar[int]
+    INT_VALUE_FIELD_NUMBER: _ClassVar[int]
+    DOUBLE_VALUE_FIELD_NUMBER: _ClassVar[int]
+    ARRAY_VALUE_FIELD_NUMBER: _ClassVar[int]
+    KVLIST_VALUE_FIELD_NUMBER: _ClassVar[int]
+    BYTES_VALUE_FIELD_NUMBER: _ClassVar[int]
+    string_value: str
+    bool_value: bool
+    int_value: int
+    double_value: float
+    array_value: ArrayValue
+    kvlist_value: KeyValueList
+    bytes_value: bytes
+    def __init__(self, string_value: _Optional[str] = ..., bool_value: bool = ..., int_value: _Optional[int] = ..., double_value: _Optional[float] = ..., array_value: _Optional[_Union[ArrayValue, _Mapping]] = ..., kvlist_value: _Optional[_Union[KeyValueList, _Mapping]] = ..., bytes_value: _Optional[bytes] = ...) -> None: ...
+
+class ArrayValue(_message.Message):
+    __slots__ = ("values",)
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    values: _containers.RepeatedCompositeFieldContainer[AnyValue]
+    def __init__(self, values: _Optional[_Iterable[_Union[AnyValue, _Mapping]]] = ...) -> None: ...
+
+class KeyValueList(_message.Message):
+    __slots__ = ("values",)
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    values: _containers.RepeatedCompositeFieldContainer[KeyValue]
+    def __init__(self, values: _Optional[_Iterable[_Union[KeyValue, _Mapping]]] = ...) -> None: ...
+
+class KeyValue(_message.Message):
+    __slots__ = ("key", "value")
+    KEY_FIELD_NUMBER: _ClassVar[int]
+    VALUE_FIELD_NUMBER: _ClassVar[int]
+    key: str
+    value: AnyValue
+    def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AnyValue, _Mapping]] = ...) -> None: ...

hebra/proto/core/collector/v1/collector_service.proto

@@ -0,0 +1,23 @@
+syntax = "proto3";
+
+package swanlab.proto.core.collector.v1;
+
+import "common/v1/common.proto";
+
+option go_package = "core/pkg/pb";
+
+
+message CollectorUploadRequest {
+  swanlab.proto.common.v1.KeyValueList data = 1;
+}
+
+message CollectorUploadResponse {
+  bool success = 1;
+  string message = 2;
+}
+
+// Collector is an RPC service responsible for aggregating and reporting data
+service Collector {
+  // Upload allows clients to send data to the collector.
+  rpc Upload(CollectorUploadRequest) returns (CollectorUploadResponse);
+}

hebra/proto/core/collector/v1/collector_service_pb2.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: core/collector/v1/collector_service.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'core/collector/v1/collector_service.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from common.v1 import common_pb2 as common_dot_v1_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)core/collector/v1/collector_service.proto\x12\x1fswanlab.proto.core.collector.v1\x1a\x16\x63ommon/v1/common.proto\"M\n\x16\x43ollectorUploadRequest\x12\x33\n\x04\x64\x61ta\x18\x01 \x01(\x0b\x32%.swanlab.proto.common.v1.KeyValueList\";\n\x17\x43ollectorUploadResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t2\x88\x01\n\tCollector\x12{\n\x06Upload\x12\x37.swanlab.proto.core.collector.v1.CollectorUploadRequest\x1a\x38.swanlab.proto.core.collector.v1.CollectorUploadResponseB\rZ\x0b\x63ore/pkg/pbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'core.collector.v1.collector_service_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\013core/pkg/pb'
+  _globals['_COLLECTORUPLOADREQUEST']._serialized_start=102
+  _globals['_COLLECTORUPLOADREQUEST']._serialized_end=179
+  _globals['_COLLECTORUPLOADRESPONSE']._serialized_start=181
+  _globals['_COLLECTORUPLOADRESPONSE']._serialized_end=240
+  _globals['_COLLECTOR']._serialized_start=243
+  _globals['_COLLECTOR']._serialized_end=379
+# @@protoc_insertion_point(module_scope)

hebra/proto/core/collector/v1/collector_service_pb2.pyi

@@ -0,0 +1,21 @@
+from common.v1 import common_pb2 as _common_pb2
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class CollectorUploadRequest(_message.Message):
+    __slots__ = ("data",)
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    data: _common_pb2.KeyValueList
+    def __init__(self, data: _Optional[_Union[_common_pb2.KeyValueList, _Mapping]] = ...) -> None: ...
+
+class CollectorUploadResponse(_message.Message):
+    __slots__ = ("success", "message")
+    SUCCESS_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    success: bool
+    message: str
+    def __init__(self, success: bool = ..., message: _Optional[str] = ...) -> None: ...

hebra/proto/core/collector/v1/collector_service_pb2_grpc.py

@@ -0,0 +1,101 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from core.collector.v1 import collector_service_pb2 as core_dot_collector_dot_v1_dot_collector__service__pb2
+
+GRPC_GENERATED_VERSION = '1.76.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + ' but the generated code in core/collector/v1/collector_service_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class CollectorStub(object):
+    """Collector is an RPC service responsible for aggregating and reporting data
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Upload = channel.unary_unary(
+                '/swanlab.proto.core.collector.v1.Collector/Upload',
+                request_serializer=core_dot_collector_dot_v1_dot_collector__service__pb2.CollectorUploadRequest.SerializeToString,
+                response_deserializer=core_dot_collector_dot_v1_dot_collector__service__pb2.CollectorUploadResponse.FromString,
+                _registered_method=True)
+
+
+class CollectorServicer(object):
+    """Collector is an RPC service responsible for aggregating and reporting data
+    """
+
+    def Upload(self, request, context):
+        """Upload allows clients to send data to the collector.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_CollectorServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Upload': grpc.unary_unary_rpc_method_handler(
+                    servicer.Upload,
+                    request_deserializer=core_dot_collector_dot_v1_dot_collector__service__pb2.CollectorUploadRequest.FromString,
+                    response_serializer=core_dot_collector_dot_v1_dot_collector__service__pb2.CollectorUploadResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'swanlab.proto.core.collector.v1.Collector', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('swanlab.proto.core.collector.v1.Collector', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class Collector(object):
+    """Collector is an RPC service responsible for aggregating and reporting data
+    """
+
+    @staticmethod
+    def Upload(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/swanlab.proto.core.collector.v1.Collector/Upload',
+            core_dot_collector_dot_v1_dot_collector__service__pb2.CollectorUploadRequest.SerializeToString,
+            core_dot_collector_dot_v1_dot_collector__service__pb2.CollectorUploadResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)