#NEWLINE# def _adjust_shape( dat , k_vars ) : #NEWLINE# #INDENT# ' Returns an array of shape (nobs, k_vars) for use with `gpke`.' #NEWLINE# dat = np . asarray ( dat ) #NEWLINE# if ( dat . ndim > 2 ) : #NEWLINE# #INDENT# dat = np . squeeze ( dat ) #NEWLINE# #UNINDENT# if ( ( dat . ndim == 1 ) and ( k_vars > 1 ) ) : #NEWLINE# #INDENT# nobs = 1 #NEWLINE# #UNINDENT# else : #NEWLINE# #INDENT# if ( ( dat . ndim == 1 ) and ( k_vars == 1 ) ) : #NEWLINE# #INDENT# nobs = len ( dat ) #NEWLINE# #UNINDENT# else : #NEWLINE# #INDENT# if ( ( np . shape ( k_vars ) [ 0 ] == k_vars ) and ( np . shape ( dat ) [ 1 ] != k_vars ) ) : #NEWLINE# #INDENT# dat = dat . T #NEWLINE# #UNINDENT# nobs = np . shape ( dat ) [ 0 ] #NEWLINE# #UNINDENT# #UNINDENT# dat = np . reshape ( dat , ( nobs , k_vars ) ) #NEWLINE# return dat