Skip to content

Commit bdc40d9

Browse files
committed
RTU get_operation_factory
1 parent 8d190ac commit bdc40d9

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

temporalio/worker/_interceptor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def __post_init__(self) -> None:
313313
self._operation_name = self.operation
314314
self._input_type = None
315315
elif isinstance(self.operation, Callable):
316-
_, op = nexusrpc.handler.get_operation_factory(self.operation)
316+
_, op = nexusrpc.get_operation_factory(self.operation)
317317
if isinstance(op, nexusrpc.Operation):
318318
self._operation_name = op.name
319319
self._input_type = op.input_type

tests/nexus/test_dynamic_creation_of_user_handler_classes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
import httpx
44
import nexusrpc.handler
55
import pytest
6+
from nexusrpc import get_operation_factory
67
from nexusrpc.handler import sync_operation
7-
from nexusrpc.handler._util import get_operation_factory
88

99
from temporalio.client import Client
1010
from temporalio.worker import Worker

tests/nexus/test_handler_operation_definitions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ async def test_collected_operation_names(
9292
assert isinstance(service_defn, nexusrpc.ServiceDefinition)
9393
assert service_defn.name == "Service"
9494
for method_name, expected_op in test_case.expected_operations.items():
95-
_, actual_op = nexusrpc.handler.get_operation_factory(
95+
_, actual_op = nexusrpc.get_operation_factory(
9696
getattr(test_case.Service, method_name)
9797
)
9898
assert isinstance(actual_op, nexusrpc.Operation)

0 commit comments

Comments
 (0)