ENH: Removed keyword from docstring that was removed from method

Author: rstoneback

OMMBV/satellite.py

@@ -1,7 +1,8 @@
 import OMMBV as OMMBV
 
 
-def add_mag_drift_unit_vectors_ecef(inst, **kwargs):
+def add_mag_drift_unit_vectors_ecef(inst, lat_label='latitude', long_label='longitude',
+                                    alt_label='altitude', **kwargs):
     """Adds unit vectors expressing the ion drift coordinate system
     organized by the geomagnetic field. Unit vectors are expressed
     in ECEF coordinates.
@@ -10,8 +11,6 @@ def add_mag_drift_unit_vectors_ecef(inst, **kwargs):
     ----------
     inst : pysat.Instrument
         Instrument object that will get unit vectors
-    max_steps : int
-        Maximum number of steps allowed for field line tracing
     **kwargs
         Passed along to calculate_mag_drift_unit_vectors_ecef
 
@@ -27,10 +26,10 @@ def add_mag_drift_unit_vectors_ecef(inst, **kwargs):
     """
 
     # add unit vectors for magnetic drifts in ecef coordinates
-    zvx, zvy, zvz, bx, by, bz, mx, my, mz = OMMBV.calculate_mag_drift_unit_vectors_ecef(inst['latitude'],
-                                                                                        inst['longitude'],
-                                                                                        inst['altitude'],
-                                                                                        inst.data.index,
+    zvx, zvy, zvz, bx, by, bz, mx, my, mz = OMMBV.calculate_mag_drift_unit_vectors_ecef(inst[lat_label],
+                                                                                        inst[long_label],
+                                                                                        inst[alt_label],
+                                                                                        inst.index,
                                                                                         **kwargs)
 
     inst['unit_zon_ecef_x'] = zvx
@@ -168,7 +167,8 @@ def add_mag_drift_unit_vectors_ecef(inst, **kwargs):
     return
 
 
-def add_mag_drift_unit_vectors(inst, **kwargs):
+def add_mag_drift_unit_vectors(inst, lat_label='latitude', long_label='longitude',
+                                    alt_label='altitude', **kwargs):
     """Add unit vectors expressing the ion drift coordinate system
     organized by the geomagnetic field. Unit vectors are expressed
     in S/C coordinates.
@@ -198,7 +198,8 @@ def add_mag_drift_unit_vectors(inst, **kwargs):
     """
 
     # vectors are returned in geo/ecef coordinate system
-    add_mag_drift_unit_vectors_ecef(inst, **kwargs)
+    add_mag_drift_unit_vectors_ecef(inst, lat_label=lat_label, long_label=long_label,
+                                    alt_label=alt_label,**kwargs)
     # convert them to S/C using transformation supplied by OA
     inst['unit_zon_x'], inst['unit_zon_y'], inst['unit_zon_z'] = OMMBV.project_ecef_vector_onto_basis(
         inst['unit_zon_ecef_x'], inst['unit_zon_ecef_y'], inst['unit_zon_ecef_z'],