fix(autogram): Reset even when exception is raised#417

ValerianRey merged 1 commit intomainSimplexLab/TorchJD:mainfrom

fix-state-reset-when-exceptionSimplexLab/TorchJD:fix-state-reset-when-exceptionCopy head branch name to clipboard