rtdl.data | fix piecewise_linear_encoding

yandex-research · Aug 9, 2023 · 60ae981 · 60ae981
1 parent 7929497
commit 60ae981
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/rtdl/data.py b/rtdl/data.py
@@ -587,7 +587,6 @@ 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_edges)
     bin_ratios = torch.as_tensor(bin_ratios)
     bin_indices = torch.as_tensor(bin_indices)