Merge pull request #691 from vlavorini/devel

Added missing columns (:), fixed indentation
This commit is contained in:
Zhenwen Dai 2018-10-30 09:52:06 +00:00 committed by GitHub
commit 9c0213335b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -484,7 +484,7 @@ cdef class AQcompute_batch_Cython(Q_handling_Cython):
if matrix_index in self.Q_square_root_dict: if matrix_index in self.Q_square_root_dict:
square_root = self.Q_square_root_dict[matrix_index] square_root = self.Q_square_root_dict[matrix_index]
else: else:
if matrix_index not in self.Q_svd_dict if matrix_index not in self.Q_svd_dict:
U,S,Vh = sp.linalg.svd( self.Qs[:,:, matrix_index], U,S,Vh = sp.linalg.svd( self.Qs[:,:, matrix_index],
full_matrices=False, compute_uv=True, full_matrices=False, compute_uv=True,
overwrite_a=False, check_finite=False) overwrite_a=False, check_finite=False)
@ -514,7 +514,7 @@ cdef class AQcompute_batch_Cython(Q_handling_Cython):
if matrix_index in self.Q_inverse_dict: if matrix_index in self.Q_inverse_dict:
Q_inverse = self.Q_inverse_dict[matrix_index] Q_inverse = self.Q_inverse_dict[matrix_index]
else: else:
if matrix_index not in self.Q_svd_dict if matrix_index not in self.Q_svd_dict:
U,S,Vh = sp.linalg.svd( self.Qs[:,:, matrix_index], U,S,Vh = sp.linalg.svd( self.Qs[:,:, matrix_index],
full_matrices=False, compute_uv=True, full_matrices=False, compute_uv=True,
overwrite_a=False, check_finite=False) overwrite_a=False, check_finite=False)
@ -522,7 +522,7 @@ cdef class AQcompute_batch_Cython(Q_handling_Cython):
else: else:
U,S,Vh = self.Q_svd_dict[matrix_index] U,S,Vh = self.Q_svd_dict[matrix_index]
Q_inverse = Q_inverse = np.dot( Vh.T * ( 1.0/(S + jitter)) , U.T ) Q_inverse = Q_inverse = np.dot( Vh.T * ( 1.0/(S + jitter)) , U.T )
self.Q_inverse_dict[matrix_index] = Q_inverse self.Q_inverse_dict[matrix_index] = Q_inverse
return Q_inverse return Q_inverse