Remove span graph module from LIT.

PiperOrigin-RevId: 644421561
PAIR-code · Jun 18, 2024 · 0656386 · 0656386
1 parent 3dad2b0
commit 0656386
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 1,285 deletions.
diff --git a/lit_nlp/api/layout.py b/lit_nlp/api/layout.py
@@ -59,10 +59,6 @@ class LitModuleName(dtypes.EnumSerializableAsValues, enum.Enum):
   SimpleDatapointEditorModule = 'simple-datapoint-editor-module'
   # Non-replicating version of Datapoint Editor
   SingleDatapointEditorModule = 'single-datapoint-editor-module'
-  SpanGraphGoldModule = 'span-graph-gold-module'
-  SpanGraphGoldModuleVertical = 'span-graph-gold-module-vertical'
-  SpanGraphModule = 'span-graph-module'
-  SpanGraphModuleVertical = 'span-graph-module-vertical'
   TCAVModule = 'tcav-module'
   ThresholderModule = 'thresholder-module'
   TrainingDataAttributionModule = 'tda-module'
@@ -126,8 +122,6 @@ def to_json(self) -> dtypes.JsonDict:
 modules = LitModuleName  # pylint: disable=invalid-name
 
 MODEL_PREDS_MODULES = (
-    modules.SpanGraphGoldModuleVertical,
-    modules.SpanGraphModuleVertical,
     modules.ClassificationModule,
     modules.MultilabelModule,
     modules.RegressionModule,

diff --git a/lit_nlp/client/elements/span_graph_vis.css b/lit_nlp/client/elements/span_graph_vis.css