726aba089d
* pytorch timesteps * style * get rid of if-else * fix test Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>