Check if function based pyro module for use_kl_weight

[justjhong]

Fixes #1241

[codecov]

Codecov Report

Merging #1242 (d8bd388) into master (85fb04a) will increase coverage by 0.00%.
The diff coverage is 100.00%.

@@           Coverage Diff           @@
##           master    #1242   +/-   ##
=======================================
  Coverage   90.68%   90.69%           
=======================================
  Files          96       96           
  Lines        7754     7759    +5     
=======================================
+ Hits         7032     7037    +5     
  Misses        722      722           

Impacted Files	Coverage Δ
scvi/train/_trainingplans.py	95.95% <100.00%> (+0.06%)	⬆️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 85fb04a...d8bd388. Read the comment docs.

[adamgayoso]

LGTM

[adamgayoso]

@meeseeksdev backport to 0.14.x

[meeseeksdev]

Um why am I tagged in this project???

…

On Thu, Oct 28, 2021 at 1:14 PM Adam Gayoso ***@***.***> wrote: @meeseeksdev <https:/meeseeksdev> backport to 0.14.x — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub <#1242 (comment)>, or unsubscribe <https:/notifications/unsubscribe-auth/AWCRU4LZGLBHS5XPPH6NRKDUJGHHFANCNFSM5GYT36EA> . Triage notifications on the go with GitHub Mobile for iOS <https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675> or Android <https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub>.

[justjhong]

@meeseeksdev is the precursor to commands to this bot: https:/MeeseeksBox/MeeseeksDev. You may want to choose a different username because I am sure others will make the same mistake.