Source code for pyreco.remove_transients



[docs]
def RemoveTransients_Res(ResStates, Transients):
    return ResStates[Transients:,:]



[docs]
def RemoveTransient_Inps(X, Transients):
    return X[:,Transients:,:]



[docs]
def RemoveTransient_Outs(Y, Transients):
    return Y[:,Transients:,:]




[docs]
def TransientRemover(What: str, ResStates, X, Y, Transients: int):

    if What=='RX':
        return RemoveTransients_Res(ResStates, Transients), RemoveTransient_Inps(X, Transients)
    if What == 'RXY':
        return RemoveTransients_Res(ResStates, Transients), RemoveTransient_Inps(X, Transients), RemoveTransient_Outs(Y, Transients)