Skip to content

Commit

Permalink
Massage the code of distribution_properties_test a bit.
Browse files Browse the repository at this point in the history
- Split the `distributions` strategy into one for `base_distributions`
  and one for `distributions` that includes compounds.
  - In so doing, increase coverage of compound distributions.
  - Also limit the nesting depth of compound distributions.
- Define a namedtuple for the information needed to instantiate
  a Distribution.
- Smaller tweaks.

PiperOrigin-RevId: 257438879
  • Loading branch information
axch authored and tensorflower-gardener committed Jul 10, 2019
1 parent 20b5547 commit 8d97f5a
Show file tree
Hide file tree
Showing 3 changed files with 149 additions and 70 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -105,9 +105,9 @@ def _constraint(param):
return draw(
tfp_hps.broadcasting_params(
batch_shape,
params_event_ndims,
event_dim=event_dim,
enable_vars=enable_vars,
params_event_ndims=params_event_ndims,
constraint_fn_for=_constraint,
mutex_params=MUTEX_PARAMS))

Expand Down
Loading

0 comments on commit 8d97f5a

Please sign in to comment.