redner_example: remove kwargs from the function

GooeyAI · Mar 7, 2025 · 3b7e2bb · 3b7e2bb
1 parent e12ae88
commit 3b7e2bb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
@@ -2275,27 +2275,28 @@ def render_example(
         self,
         published_run: PublishedRun,
         use_workspace_author: bool = False,
-        **kwargs,
+        workflow_pill: str | None = None,
+        hide_visibility_pill: bool = False,
     ):
         max_desc_words = 280
         max_desc_words_mobile = 100
         tb = get_title_breadcrumbs(self, published_run.saved_run, published_run)
         version = published_run.versions.latest()
         pills = [
             lambda: (
-                gui.pill(
+                None
+                if hide_visibility_pill
+                else gui.pill(
                     PublishedRunVisibility(published_run.visibility).get_badge_html(),
                     unsafe_allow_html=True,
                     className="border border-dark",
                 )
-                if not "hide_visibility_pill" in kwargs
-                else None
             ),
         ]
-        if kwargs.get("workflow_pill"):
+        if workflow_pill:
             pills.append(
                 lambda: gui.pill(
-                    kwargs["workflow_pill"],
+                    workflow_pill,
                     unsafe_allow_html=True,
                     className="border border-dark ms-2",
                 )

diff --git a/routers/account.py b/routers/account.py
@@ -293,7 +293,6 @@ def all_saved_runs_tab(request: Request):
 
     def _render_run(pr: PublishedRun):
         workflow = Workflow(pr.workflow)
-        visibility = PublishedRunVisibility(pr.visibility)
 
         workflow.page_cls().render_example(pr, workflow_pill=workflow.short_title)