Automated Code Change

PiperOrigin-RevId: 772786015

Author: Googler

lit_nlp/client/modules/annotated_text_module.ts

@@ -19,12 +19,15 @@ import {customElement} from 'lit/decorators.js';
 import {makeObservable, observable} from 'mobx';
 
 import {LitModule} from '../core/lit_module';
-import {type AnnotationGroups, TextSegments} from '../elements/annotated_text_vis';
+import {type AnnotationGroups, type AnnotationSpec, type SegmentSpec, TextSegments} from '../elements/annotated_text_vis';
 import {MultiSegmentAnnotations, TextSegment} from '../lib/lit_types';
 import {styles as sharedStyles} from '../lib/shared_styles.css';
 import {type IndexedInput, ModelInfoMap, Spec} from '../lib/types';
 import {doesOutputSpecContain, filterToKeys, findSpecKeys} from '../lib/utils';
 
+// This should be removed.
+type AnyDuringMigration = any;
+
 /** LIT module for model output. */
 @customElement('annotated-text-gold-module')
 export class AnnotatedTextGoldModule extends LitModule {
@@ -53,13 +56,15 @@ export class AnnotatedTextGoldModule extends LitModule {
     // Text segment fields
     const segmentNames = findSpecKeys(dataSpec, TextSegment);
     const segments: TextSegments = filterToKeys(input.data, segmentNames);
-    const segmentSpec = filterToKeys(dataSpec, segmentNames);
+    const segmentSpec: SegmentSpec =
+        filterToKeys(dataSpec, segmentNames) as AnyDuringMigration;
 
     // Annotation fields
     const annotationNames = findSpecKeys(dataSpec, MultiSegmentAnnotations);
     const annotations: AnnotationGroups =
         filterToKeys(input.data, annotationNames);
-    const annotationSpec = filterToKeys(dataSpec, annotationNames);
+    const annotationSpec: AnnotationSpec =
+        filterToKeys(dataSpec, annotationNames) as AnyDuringMigration;
 
     // If more than one model is selected, AnnotatedTextModule will be offset
     // vertically due to the model name header, while this one won't be.
@@ -149,12 +154,15 @@ export class AnnotatedTextModule extends LitModule {
         findSpecKeys(this.appState.currentDatasetSpec, TextSegment);
     const segments: TextSegments =
         filterToKeys(this.currentData.data, segmentNames);
-    const segmentSpec =
-        filterToKeys(this.appState.currentDatasetSpec, segmentNames);
+    const segmentSpec: SegmentSpec =
+        filterToKeys(this.appState.currentDatasetSpec, segmentNames) as
+        AnyDuringMigration;
 
     const outputSpec = this.appState.getModelSpec(this.model).output;
-    const annotationSpec = filterToKeys(
-        outputSpec, findSpecKeys(outputSpec, MultiSegmentAnnotations));
+    const annotationSpec: AnnotationSpec =
+        filterToKeys(
+            outputSpec, findSpecKeys(outputSpec, MultiSegmentAnnotations)) as
+        AnyDuringMigration;
     // clang-format off
     return html`
       <annotated-text-vis .segments=${segments}

lit_nlp/client/modules/feature_attribution_module.ts

@@ -30,14 +30,14 @@ import {LegendType} from '../elements/color_legend';
 import {InterpreterClick, InterpreterSettings} from '../elements/interpreter_controls';
 import {SortableTemplateResult, TableData} from '../elements/table';
 import {FeatureSalience as FeatureSalienceLitType, LitTypeWithVocab, SingleFieldMatcher} from '../lib/lit_types';
-import {IndexedInput, ModelInfoMap} from '../lib/types';
+import {styles as sharedStyles} from '../lib/shared_styles.css';
+import {type D3Scale, IndexedInput, ModelInfoMap} from '../lib/types';
 import * as utils from '../lib/utils';
 import {findSpecKeys} from '../lib/utils';
 import {SignedSalienceCmap} from '../services/color_service';
 import {type NumericFeatureBins} from '../services/group_service';
 import {AppState, GroupService} from '../services/services';
 
-import {styles as sharedStyles} from '../lib/shared_styles.css';
 import {styles} from './feature_attribution_module.css';
 
 const ALL_DATA = 'Entire Dataset';
@@ -75,6 +75,9 @@ interface VisToggles {
   [name: string]: boolean;
 }
 
+// This should be removed.
+type AnyDuringMigration = any;
+
 /** Aggregate feature attribution for tabular ML models. */
 @customElement('feature-attribution-module')
 export class FeatureAttributionModule extends LitModule {
@@ -440,7 +443,8 @@ export class FeatureAttributionModule extends LitModule {
   }
 
   override renderImpl() {
-    const scale = (val: number) => this.colorMap.bgCmap(val);
+    const scale: D3Scale =
+        ((val: number) => this.colorMap.bgCmap(val)) as AnyDuringMigration;
     scale.domain = () => this.colorMap.colorScale.domain();
 
     // clang-format off

lit_nlp/client/modules/pdp_module.ts

@@ -44,6 +44,9 @@ interface AllPdpInfo {
 // Data for bar or line charts.
 type ChartInfo = Map<string|number, number>;
 
+// This should be removed.
+type AnyDuringMigration = any;
+
 /**
  * A LIT module that renders regression results.
  */
@@ -172,14 +175,16 @@ export class PdpModule extends LitModule {
       const yRange = isClassification ? [0, 1] : [];
       const renderChart = (chartData: ChartInfo) => {
         if (isNumeric) {
+          const chartMap: Map<number, number> = chartData as AnyDuringMigration;
           return html`
               <line-chart height=150 width=300
-                  .scores=${chartData} .yScale=${yRange}>
+                  .scores=${chartMap} .yScale=${yRange}>
               </line-chart>`;
         } else {
+          const chartMap: Map<string, number> = chartData as AnyDuringMigration;
           return html`
               <bar-chart height=150 width=300
-                  .scores=${chartData} .yScale=${yRange}>
+                  .scores=${chartMap} .yScale=${yRange}>
               </bar-chart>`;
         }