fix typeerror 'NoneType' object is not iterable

GooeyAI · Jul 9, 2024 · 0ef97f1 · 0ef97f1
1 parent 0ddcd34
commit 0ef97f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/daras_ai_v2/prompt_vars.py b/daras_ai_v2/prompt_vars.py
@@ -32,7 +32,7 @@ def variables_input(
         else:
             template_var_names |= jinja2.meta.find_undeclared_variables(parsed)
 
-    old_vars = st.session_state.get(key, {})
+    old_vars = st.session_state.get(key) or {}
 
     var_add_key = f"--{key}:add_btn"
     var_name_key = f"--{key}:add_name"