Skip to content

Commit ff0413d

Browse files
committed
Fix python 3.8 and 3.9 tests
1 parent 08f736c commit ff0413d

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

instrumentation-genai/opentelemetry-instrumentation-vertexai/tests/test_chat_completions.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -470,19 +470,20 @@ def generate_content_all_input_events(
470470
}
471471

472472

473-
# Type annotation for fixture to make LSP work properly
474-
class GenerateContentFixture(Protocol):
475-
_P = ParamSpec("_P")
476-
_R = TypeVar("_R")
473+
_P = ParamSpec("_P")
474+
_R = TypeVar("_R")
475+
477476

478-
@staticmethod
479-
def _copy_signature(
480-
func_type: Callable[_P, _R],
481-
) -> Callable[
482-
[Callable[..., Any]], Callable[Concatenate[GenerativeModel, _P], _R]
483-
]:
484-
return lambda func: func
477+
def _copy_signature(
478+
func_type: Callable[_P, _R],
479+
) -> Callable[
480+
[Callable[..., Any]], Callable[Concatenate[GenerativeModel, _P], _R]
481+
]:
482+
return lambda func: func
485483

484+
485+
# Type annotation for fixture to make LSP work properly
486+
class GenerateContentFixture(Protocol):
486487
@_copy_signature(GenerativeModel.generate_content)
487488
def __call__(self): ...
488489

0 commit comments

Comments
 (0)