Fix bug in add_points method

* solving `add_points`

---------

Co-authored-by: Gabriele Codega <gcodega@pascal.maths.sissa.it>
This commit is contained in:
Gabriele Codega
2024-06-20 12:57:46 +02:00
committed by GitHub
parent 35fd119a37
commit c7973a97da
2 changed files with 13 additions and 2 deletions

View File

@@ -273,7 +273,7 @@ class AbstractProblem(metaclass=ABCMeta):
new_pts.labels = old_pts.labels
# merging
merged_pts = torch.vstack([old_pts, new_points[location]])
merged_pts = torch.vstack([old_pts, new_pts])
merged_pts.labels = old_pts.labels
self.input_pts[location] = merged_pts