From a4afbfd7391de4b5df70558a63e7fe7281a82424 Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Sun, 3 Oct 2021 10:27:34 -0700
Subject: [PATCH] Small fixes to PMobject

---
 manimlib/mobject/types/point_cloud_mobject.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/manimlib/mobject/types/point_cloud_mobject.py b/manimlib/mobject/types/point_cloud_mobject.py
index 844641dd..809c2248 100644
--- a/manimlib/mobject/types/point_cloud_mobject.py
+++ b/manimlib/mobject/types/point_cloud_mobject.py
@@ -20,6 +20,11 @@ class PMobject(Mobject):
                 self.data[key] = resize_array(self.data[key], size)
         return self
 
+    def set_points(self, points):
+        super().set_points(points)
+        self.resize_points(len(points))
+        return self
+
     def add_points(self, points, rgbas=None, color=None, opacity=None):
         """
         points must be a Nx3 numpy array, as must rgbas if it is not None
@@ -87,6 +92,11 @@ class PMobject(Mobject):
         lower_index = int(a * pmobject.get_num_points())
         upper_index = int(b * pmobject.get_num_points())
         for key in self.data:
+            if key == "bounding_box":
+                continue
+            # data_len = len(self.data[key])
+            # lower_index = int(a * data_len)
+            # upper_index = int(np.ceil(b * data_len))
             self.data[key] = pmobject.data[key][lower_index:upper_index]
         return self