fix typo in PD

Tractables · Mar 6, 2024 · e867fd7 · e867fd7
1 parent d6e5204
commit e867fd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pyjuice/structures/pd.py b/src/pyjuice/structures/pd.py
@@ -21,7 +21,7 @@ def PD(data_shape: Tuple, num_latents: int,
        max_prod_block_conns: int = 4,
        structure_type: str = "sum_dominated",
        input_layer_fn: Optional[Callable] = None,
-       input_dists: Optional[Distribution] = None,
+       input_dist: Optional[Distribution] = None,
        input_layer_type: Type[Distribution] = Categorical, 
        input_layer_params: Dict = {"num_cats": 256},
        use_linear_mixing: bool = False,