[Performance] Avoid computing log-probs when retrieving dist#3081

vmoens merged 1 commit intomainpytorch/rl:mainfrom

refactor-get-distpytorch/rl:refactor-get-distCopy head branch name to clipboard