Describe the purpose of the pull request, including:
Now the user can decide if the distribution arguments should be validated or not.
Removed extension from Distribution for classes: MSEDistribution, SymlogDistribution, and TwoHotEncodingDistribution.
Update docs
Added custom OneHotCategorical and OneHotCategoricalStraightThrough distributions to be able to disable the validate args in their inner categorical distribution.
Fix tests
Type of Change
Please select the one relevant option below:
New feature (non-breaking change that adds functionality)
Documentation update
Checklist
Please confirm that the following tasks have been completed:
[x] I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
[x] I have added unit tests for my changes, or updated existing tests if necessary.
[x] I have updated the documentation, if applicable.
[x] I have installed pre-commit and run locally for my code changes.
Summary
Describe the purpose of the pull request, including:
MSEDistribution
,SymlogDistribution
, andTwoHotEncodingDistribution
.OneHotCategorical
andOneHotCategoricalStraightThrough
distributions to be able to disable thevalidate args
in their inner categorical distribution.Type of Change
Please select the one relevant option below:
Checklist
Please confirm that the following tasks have been completed: