rename semantic_similarity (#1286)

evidentlyai · Sep 6, 2024 · 2bb5e6d · 2bb5e6d
1 parent fb3f7fc
commit 2bb5e6d
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/src/evidently/descriptors/semantic_similarity.py b/src/evidently/descriptors/semantic_similarity.py
@@ -1,18 +1,9 @@
-from typing import List
-
 from evidently.features.generated_features import FeatureDescriptor
-from evidently.features.generated_features import GeneratedFeature
 from evidently.features.generated_features import GeneratedFeatures
-from evidently.features.generated_features import MultiColumnFeatureDescriptor
 from evidently.features.semantic_similarity_feature import SemanticSimilarityFeature
 
 
-class SemanticSimilarity(MultiColumnFeatureDescriptor):
-    def feature(self, columns: List[str]) -> GeneratedFeature:
-        return SemanticSimilarityFeature(columns=columns, display_name=self.display_name)
-
-
-class SemanticSimilarityDescriptor(FeatureDescriptor):
+class SemanticSimilarity(FeatureDescriptor):
     with_column: str
 
     def feature(self, column_name: str) -> GeneratedFeatures: