Restrict orientation to airborne for now

Author: domfournier

simpeg_drivers/components/factories/receiver_factory.py

@@ -24,6 +24,7 @@
 
 import numpy as np
 from geoapps_utils.utils.transformations import x_rotation_matrix, z_rotation_matrix
+from geoh5py.objects.surveys.electromagnetics.base import AirborneEMSurvey
 
 from simpeg_drivers.components.factories.simpeg_factory import SimPEGFactory
 from simpeg_drivers.utils.regularization import direction_and_dip, get_cell_normals
@@ -162,7 +163,10 @@ def assemble_keyword_arguments(
             kwargs["data_type"] = "ppm"
 
         # Overload orientation if provided
-        if self.factory_type in ["tdem", "fdem"] and local_indices is not None:
+        if (
+            isinstance(self.params.data_object, AirborneEMSurvey)
+            and local_indices is not None
+        ):
             kwargs["orientation"] = self.orientations[kwargs["orientation"]][
                 local_indices, :
             ]