Made df_attr consistent

DOI-USGS · Aug 21, 2019 · bec1585 · bec1585
1 parent 8904cb9
commit bec1585
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plio/io/io_controlnetwork.py b/plio/io/io_controlnetwork.py
@@ -264,8 +264,8 @@ def _set_pid(pointid):
                 df_attr = self.point_field_map.get(attr, attr)
                 if df_attr in g.columns:
                     # As per protobuf docs for assigning to a repeated field.
-                    if attr == 'aprioriCovar' or attr == 'adjustedCovar':
-                        arr = g.iloc[0][attr]
+                    if df_attr == 'aprioriCovar' or df_attr == 'adjustedCovar':
+                        arr = g.iloc[0][df_attr]
                         if isinstance(arr, np.ndarray):
                             arr = arr.ravel().tolist()