Internal changes

PiperOrigin-RevId: 696998709 Change-Id: I18ce6063fd9195e2d1424b662427417241bfe846
google · Nov 15, 2024 · c66d36d · c66d36d
1 parent 82737fd
commit c66d36d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/saxml/server/servable_model_params.py b/saxml/server/servable_model_params.py
@@ -117,6 +117,8 @@ def sax_registration_name(cls) -> Optional[str]:
     """Returns an optional custom registration name for the model."""
     return None
 
+  # TODO(jwtan): Make this a classmethod as most of the config attributes are
+  # class attributes.
   def apply_model_overrides(self, overrides: Dict[str, Any]) -> None:
     """Applies model config overrides received from Publish.