refactor: change modality to video

Signed-off-by: Snehil Shah <[email protected]>

src/core/operators/cluster_embeddings.py

@@ -89,7 +89,7 @@ def run(input_data, n_clusters=None, modality='audio'):
         dict: A dictionary mapping cluster labels to corresponding array of payloads
 
     Raises:
-        ValueError: Modality should be either `audio` or `visual`
+        ValueError: Modality should be either `audio` or `video`
         KeyError: Each data point in input must have `embedding` and `payload` properties
     """
     # Parse data:
@@ -103,10 +103,10 @@ def run(input_data, n_clusters=None, modality='audio'):
         n_clusters = int(n_clusters) # cast it to int
         if modality == 'audio':
             labels = KMeans_clustering(matrix=matrix, n_clusters=n_clusters)
-        elif modality == 'visual':
+        elif modality == 'video':
             labels = Agglomerative_clustering(matrix=matrix, n_clusters=n_clusters)
         else:
-            raise ValueError("Invalid modality. Modality should be either `audio` or `visual`.")
+            raise ValueError("Invalid modality. Modality should be either `audio` or `video`.")
     else:
         labels = AffinityPropagation_clustering(matrix=matrix)
     return gen_data(payloads=payloads, labels=labels) # format output

src/core/operators/test_cluster_embeddings.py

@@ -30,7 +30,7 @@ def test_kmeans_clustering(self):
         self.assertCountEqual([result["cluster_0"], result["cluster_1"]], EXPECTED_CLUSTERS)
 
     def test_agglomerative_clustering(self):
-        result = cluster_embeddings.run(input_data=MOCK_DATA, n_clusters=2, modality="visual")
+        result = cluster_embeddings.run(input_data=MOCK_DATA, n_clusters=2, modality="video")
         self.assertIn("cluster_0", result)
         self.assertIn("cluster_1", result)
         self.assertEqual(len(result), 2)