From 8daec8f3993465435fbf185a99fc8ac6ac16ec32 Mon Sep 17 00:00:00 2001 From: ephoris Date: Mon, 30 Sep 2024 14:05:03 -0400 Subject: [PATCH] [Bug] Fix error in merging --- endure/lcm/util/util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/endure/lcm/util/util.py b/endure/lcm/util/util.py index 6aa7284..b9931ee 100644 --- a/endure/lcm/util/util.py +++ b/endure/lcm/util/util.py @@ -63,7 +63,7 @@ def create_input_from_types( out = one_hot_lcm_classic_distinct(data, categories, size_ratio, policy) elif design.policy == Policy.KHybrid: ks = [k - 1 if k > 0 else 0 for k in design.K] - inputs = wl + sys + [design.h, size_ratio] + ks + inputs = wl + sys + [design.h, size_ratio_idx] + ks data = torch.Tensor(inputs) num_feats = 1 + len(design.K) out = one_hot_lcm(data, len(inputs), num_feats, categories)