Add chart options to fsapi (watertap-org#1413)

* add chart_type and chart_group to modelexport * update test to include chart_type and chart_group --------- Co-authored-by: Ludovico Bianchi <lbianchi@lbl.gov> Co-authored-by: Adam Atia <aatia@keylogic.com>
agarciadiego · Jun 6, 2024 · 6ef2765 · 6ef2765
1 parent 3d42a54
commit 6ef2765
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/watertap/ui/fsapi.py b/watertap/ui/fsapi.py
@@ -92,6 +92,8 @@ class ModelExport(BaseModel):
     is_readonly: Union[None, bool] = Field(default=None, validate_default=True)
     input_category: Optional[str] = None
     output_category: Optional[str] = None
+    chart_type: Optional[str] = None
+    chart_group: Optional[str] = None
     # computed
     obj_key: Union[None, str] = Field(default=None, validate_default=True)
     fixed: bool = True

diff --git a/watertap/ui/tests/test_fsapi.py b/watertap/ui/tests/test_fsapi.py
@@ -97,6 +97,8 @@ def export_to_ui(flowsheet=None, exports=None, build_options=None, **kwargs):
         read_only=False,
         is_output=True,
         output_category=OutputCategory.feed,
+        chart_type="stacked_bar",
+        chart_group="test",
     )