From c66d36d2148c28951da202f43ee1036673da75e9 Mon Sep 17 00:00:00 2001
From: Jiewen Tan <jwtan@google.com>
Date: Fri, 15 Nov 2024 14:17:26 -0800
Subject: [PATCH] Internal changes

PiperOrigin-RevId: 696998709
Change-Id: I18ce6063fd9195e2d1424b662427417241bfe846
---
 saxml/server/servable_model_params.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/saxml/server/servable_model_params.py b/saxml/server/servable_model_params.py
index 9a42ba81..8872afbf 100644
--- 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.