Remove intermediate variables

equinor · Oct 3, 2024 · fa529d2 · fa529d2
1 parent e42bc49
commit fa529d2
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/ert/config/observation_vector.py b/src/ert/config/observation_vector.py
@@ -31,8 +31,6 @@ def __len__(self) -> int:
 
     def to_dataset(self, active_list: List[int]) -> polars.DataFrame:
         if self.observation_type == EnkfObservationImplementationType.GEN_OBS:
-            actual_response_key = self.data_key
-            actual_observation_key = self.observation_key
             dataframes = []
             for time_step, node in self.observations.items():
                 if active_list and time_step not in active_list:
@@ -42,8 +40,8 @@ def to_dataset(self, active_list: List[int]) -> polars.DataFrame:
                 dataframes.append(
                     polars.DataFrame(
                         {
-                            "response_key": actual_response_key,
-                            "observation_key": actual_observation_key,
+                            "response_key": self.data_key,
+                            "observation_key": self.observation_key,
                             "report_step": polars.Series(
                                 np.full(len(node.indices), time_step),
                                 dtype=polars.UInt16,