Update domain_space.py

domain_space.py

@@ -205,13 +205,14 @@ def make_domain_plot(df_synth_pred, explored_domain_space, x_list, target="y_pre
 
     cols_for_plot = [c for c in df_synth_pred.columns if c not in filter_cols + ["Temperature_C"]]
     x_col, y_col, z_col = cols_for_plot[0], cols_for_plot[1], cols_for_plot[2]
+    hover_data={target: ":.3f", "uncertainty": ":.3f"}
     fig = px.scatter_3d(
         df_synth_pred2,
         x=x_col,
         y=y_col,
         z=z_col,
         color=target,
-        size=df_synth_pred["uncertainty"],
+        size=hover_data["uncertainty"],
         range_color=[uncertainty_min, uncertainty_max],
         hover_data={target: ":.3f", "uncertainty": ":.3f"},
     )