rtdl.data | fix a typo in piecewise_linear_encoding

close #47 #54
yandex-research · Aug 7, 2023 · 7929497 · 7929497
1 parent d00d89a
commit 7929497
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rtdl/data.py b/rtdl/data.py
@@ -587,7 +587,7 @@ def piecewise_linear_encoding(
             X_ple = piecewise_linear_encoding(bin_edges, bin_indices, bin_ratios, bin_counts, stack=True)
     """
     is_torch = isinstance(bin_ratios, Tensor)
-    bin_edges = torch.as_tensor(bin_ratios)
+    bin_edges = torch.as_tensor(bin_edges)
     bin_ratios = torch.as_tensor(bin_ratios)
     bin_indices = torch.as_tensor(bin_indices)