Skip to content

Commit

Permalink
Remove unsused get_proto from ConcreteComputation
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 731003356
  • Loading branch information
michaelreneer authored and copybara-github committed Feb 25, 2025
1 parent eb01087 commit d21797d
Showing 1 changed file with 6 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -266,10 +266,12 @@ def after_merge_computation(merge_result):
annotated_type_signature = federated_language.FunctionType(
after_merge_computation.type_signature.parameter, original_return_type
)
after_merge_computation = (
federated_language.framework.ConcreteComputation.with_type(
after_merge_computation, annotated_type_signature
)
# Note: Update to latest version of `federated-language` to use a public
# property and remove the pylint directive.
after_merge_computation = federated_language.framework.ConcreteComputation(
computation_proto=after_merge_computation.to_proto(),
context_stack=after_merge_computation._context_stack, # pylint: disable=protected-access
annotated_type=annotated_type_signature,
)

return mergeable_comp_execution_context.MergeableCompForm(
Expand Down

0 comments on commit d21797d

Please sign in to comment.