.

Lightning-AI · Dec 20, 2022 · f49b933 · f49b933
1 parent 0c5bfa5
commit f49b933
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lightning_app/components/serve/auto_scaler.py b/src/lightning_app/components/serve/auto_scaler.py
@@ -352,8 +352,8 @@ def _get_sample_dict_from_datatype(datatype: Any) -> dict:
             # not a pydantic model
             raise TypeError(f"datatype must be a pydantic model, for the UI to be generated. but got {datatype}")
 
-        if hasattr(datatype, "_get_sample_data"):
-            return datatype._get_sample_data()
+        if hasattr(datatype, "get_sample_data"):
+            return datatype.get_sample_data()
 
         datatype_props = datatype.schema()["properties"]
         out: Dict[str, Any] = {}