revert changes

Author: omkar-334

operators/cluster_embeddings/cluster_embeddings.py

@@ -14,10 +14,6 @@ class ClusterEmbeddings(Operator):
     def __init__(self) -> None:
         """Initialize the `ClusterEmbeddings` operator."""
         super().__init__()
-        self.matrix = []
-        self.payloads = []
-        self.modality = None
-        self.n_clusters = None
 
     @staticmethod
     def gen_data(payloads: list[dict], labels: np.ndarray) -> dict:
@@ -134,8 +130,8 @@ def run(
 
     def cleanup(self) -> None:
         """Clean up resources used by the operator."""
-        self.matrix = []
-        self.payloads = []
+        self.matrix.clear()
+        self.payloads.clear()
         self.modality = None
         self.n_clusters = None
 

operators/vid_vec_rep_clip/vid_vec_rep_clip.py

@@ -20,8 +20,6 @@ class VidVecRepClip(Operator):
     def __init__(self) -> None:
         """Initialize the `VidVecRepClip` class."""
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.model = None
-        self.processor = None
         self.frame_images = []
         self.feature_matrix = None
         self.load_model()
@@ -201,7 +199,7 @@ def cleanup(self) -> None:
         del self.processor
 
         self.frame_images.clear()
-        self.feature_matrix = None
+        self.feature_matrix.clear()
 
         gc.collect()
         if torch.cuda.is_available():
@@ -218,7 +216,6 @@ def state(self) -> dict:
             "model": self.model,
             "processor": self.processor,
             "frame_images": self.frame_images.copy(),
-            "feature_matrix": self.feature_matrix.clone().cpu().tolist()
-            if self.feature_matrix is not None
-            else [],
+            "feature_matrix": self.feature_matrix.copy() if self.feature_matrix else [],
+
         }