Fix bottomup inference

talmolab · Jan 15, 2025 · 6a7d28a · 6a7d28a
1 parent c6931a4
commit 6a7d28a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sleap_nn/inference/paf_grouping.py b/sleap_nn/inference/paf_grouping.py
@@ -184,7 +184,7 @@ def make_line_subs(
     Y = torch.cat(
         (src_peaks[:, 1].unsqueeze(dim=-1), dst_peaks[:, 1].unsqueeze(dim=-1)), dim=-1
     ).to(torch.float64)
-    samples = torch.Tensor([0, 1], device=X.device).repeat(n_candidates, 1)
+    samples = torch.tensor([0, 1], device=X.device).repeat(n_candidates, 1)
     samples_new = torch.linspace(0, 1, steps=n_line_points, device=X.device).repeat(
         n_candidates, 1
     )