Fix Integration tests

operators/cluster_embeddings/cluster_embeddings.py

@@ -14,6 +14,10 @@ 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:
@@ -130,8 +134,8 @@ def run(
 
     def cleanup(self) -> None:
         """Clean up resources used by the operator."""
-        self.matrix.clear()
-        self.payloads.clear()
+        self.matrix = []
+        self.payloads = []
         self.modality = None
         self.n_clusters = None
 

operators/vid_vec_rep_clip/vid_vec_rep_clip.py

@@ -20,8 +20,10 @@ 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 = []
+        self.feature_matrix = None
         self.load_model()
         self.validate_system()
 
@@ -199,7 +201,7 @@ def cleanup(self) -> None:
         del self.processor
 
         self.frame_images.clear()
-        self.feature_matrix.clear()
+        self.feature_matrix = None
 
         gc.collect()
         if torch.cuda.is_available():
@@ -216,5 +218,7 @@ def state(self) -> dict:
             "model": self.model,
             "processor": self.processor,
             "frame_images": self.frame_images.copy(),
-            "feature_matrix": self.feature_matrix.copy() if self.feature_matrix else [],
+            "feature_matrix": self.feature_matrix.clone().cpu().tolist()
+            if self.feature_matrix is not None
+            else [],
         }