Skip to content

Add __call__ and encapsulate CHANNEL_CONFIG in ComponentTransportModel. - #2359

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_963462268
Open

Add __call__ and encapsulate CHANNEL_CONFIG in ComponentTransportModel.#2359
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_963462268

Conversation

@copybara-service

Copy link
Copy Markdown

Add call and encapsulate CHANNEL_CONFIG in ComponentTransportModel.

  • Move CHANNEL_CONFIG_STRUCT from module-level constant to ClassVar (CHANNEL_CONFIG) on ComponentTransportModel.
  • Add call method to ComponentTransportModel wrapping call_implementation() and zero_out_disabled_channels().
  • Update zero_out_disabled_channels to use self.CHANNEL_CONFIG.
  • Update transport_model.py shim to re-export CHANNEL_CONFIG as CHANNEL_CONFIG_STRUCT.

- Move CHANNEL_CONFIG_STRUCT from module-level constant to ClassVar (CHANNEL_CONFIG) on ComponentTransportModel.
- Add __call__ method to ComponentTransportModel wrapping call_implementation() and zero_out_disabled_channels().
- Update zero_out_disabled_channels to use self.CHANNEL_CONFIG.
- Update transport_model.py shim to re-export CHANNEL_CONFIG as CHANNEL_CONFIG_STRUCT.

PiperOrigin-RevId: 963462268
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant