#NEWLINE# def add_data( self , data ) : #NEWLINE# #INDENT# assert ( isinstance ( data , np . ndarray ) and ( data . ndim == 2 ) and ( data . shape [ 1 ] == self . K ) ) #NEWLINE# T = data . shape [ 0 ] #NEWLINE# augmented_data = { #NEWLINE# #INDENT# 'x' : data , #NEWLINE# 'T' : T , #NEWLINE# #UNINDENT# } #NEWLINE# augmented_data [ 'states' ] = MultinomialLDSStates ( model = self , data = data ) #NEWLINE# self . emission_distn . augment_data ( augmented_data ) #NEWLINE# self . data_list . append ( augmented_data )