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: