| 29 | |
| 30 | @flax.struct.dataclass |
| 31 | class EncoderNetworks: |
| 32 | policy_network: networks.FeedForwardNetwork |
| 33 | prior_network: networks.FeedForwardNetwork |
| 34 | parametric_latent_distribution: distribution.ParametricDistribution |
| 35 | conditional: bool |
| 36 | random: bool |
| 37 | |
| 38 | |
| 39 | def make_inference_fn(encoder_networks: EncoderNetworks): |
no outgoing calls
no test coverage detected