Make a copy of reponses dataframe instead of using it directly

equinor · Mar 22, 2021 · f4d7dd1 · f4d7dd1
1 parent e4904df
commit f4d7dd1
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/ertviz/controllers/response_correlation_controller.py b/ertviz/controllers/response_correlation_controller.py
@@ -162,18 +162,19 @@ def update_response_overview_plot(
                 style = deepcopy(assets.ERTSTYLE["response-plot"]["response-index"])
             else:
                 style = deepcopy(assets.ERTSTYLE["response-plot"]["response"])
-            data_df = response.data_df()
+            data_df = response.data_df().copy()
 
-            style["marker"]["color"] = data["color"]
+            style.update({"marker": {"color": data["color"]}})
+            style.update({"line": {"color": data["color"]}})
             _plots += [
                 PlotModel(
                     x_axis=x_axis,
-                    y_axis=data_df[x_val],
-                    text="Mean",
-                    name=f"Mean {repr(ensemble)}",
+                    y_axis=data_df[realization],
+                    text=realization,
+                    name=realization,
                     **style,
                 )
-                for x_val in data_df
+                for realization in data_df
             ]
 
             for obs in response.observations: