update Jtvec to work with Rx classes

simpeg · May 8, 2016 · 0a71466 · 0a71466
1 parent cb042ac
commit 0a71466
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/SimPEG/EM/FDEM/FDEM.py b/SimPEG/EM/FDEM/FDEM.py
@@ -137,10 +137,9 @@ def Jtvec(self, m, v, f=None):
                     df_dmT = df_dmT + du_dmT
 
                     # TODO: this should be taken care of by the reciever?
-                    real_or_imag = rx.projComp
-                    if real_or_imag is 'real':
+                    if rx.real_or_imag is 'real':
                         Jtv +=   np.array(df_dmT, dtype=complex).real
-                    elif real_or_imag is 'imag':
+                    elif rx.real_or_imag is 'imag':
                         Jtv += - np.array(df_dmT, dtype=complex).real
                     else:
                         raise Exception('Must be real or imag')