From dfed5db111bd11d9571b661d4513afda77f15841 Mon Sep 17 00:00:00 2001
From: Steven Braun <steven.braun.mz@gmail.com>
Date: Thu, 2 Nov 2023 16:39:05 +0100
Subject: [PATCH] data: Fix get_distribution call

---
 models_pl.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models_pl.py b/models_pl.py
index 48ab5a0..e5d159e 100644
--- a/models_pl.py
+++ b/models_pl.py
@@ -31,7 +31,7 @@ def make_einet(cfg, num_classes: int = 1) -> EinetMixture | Einet:
 
     image_shape = get_data_shape(cfg.dataset)
     # leaf_kwargs, leaf_type = {"total_count": 255}, Binomial
-    leaf_kwargs, leaf_type = get_distribution(dist=cfg.dist, min_sigma=cfg.min_sigma, max_sigma=cfg.max_sigma)
+    leaf_kwargs, leaf_type = get_distribution(dist=cfg.dist, cfg=cfg)
 
     config = EinetConfig(
         num_features=image_shape.num_pixels,