fixed unbound local error

wjm41 · Mar 30, 2023 · 71f5e9b · 71f5e9b
1 parent 680404a
commit 71f5e9b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/molplotly/main.py b/molplotly/main.py
@@ -246,6 +246,11 @@ def add_molecules(
     else:
         colors = {0: "black"}
 
+    if not svg_height:
+        svg_height = svg_size
+    if not svg_width:
+        svg_width = svg_size
+
     app = JupyterDash(__name__)
     if smiles_col is None and mol_col is None:
         raise ValueError("Either smiles_col or mol_col has to be specified!")
@@ -349,10 +354,6 @@ def display_hover(hoverData, value):
             df_row = df.iloc[num]
 
         hoverbox_elements = []
-        if not svg_height:
-            svg_height = svg_size
-        if not svg_width:
-            svg_width = svg_size
 
         if show_img:
             for col in chosen_smiles: