@@ -32,7 +32,6 @@ class LitModuleName(dtypes.EnumSerializableAsValues, enum.Enum):
3232 # keep-sorted start
3333 AnnotatedTextGoldModule = 'annotated-text-gold-module'
3434 AnnotatedTextModule = 'annotated-text-module'
35- AttentionModule = 'attention-module'
3635 ClassificationModule = 'classification-module'
3736 ConfusionMatrixModule = 'confusion-matrix-module'
3837 CurvesModule = 'curves-module'
@@ -198,7 +197,6 @@ def to_json(self) -> dtypes.JsonDict:
198197 'Explanations' : [
199198 modules .SalienceMapModule ,
200199 modules .LegacySequenceSalienceModule ,
201- modules .AttentionModule ,
202200 modules .FeatureAttributionModule ,
203201 ],
204202 'Clustering' : [modules .SalienceClusteringModule ],
@@ -232,7 +230,6 @@ def to_json(self) -> dtypes.JsonDict:
232230 * MODEL_PREDS_MODULES ,
233231 modules .SalienceMapModule ,
234232 modules .LegacySequenceSalienceModule ,
235- modules .AttentionModule ,
236233 modules .FeatureAttributionModule ,
237234 ],
238235 'Salience Clustering' : [modules .SalienceClusteringModule ],
0 commit comments