@@ -25,7 +25,7 @@ import { Codicon } from "../common";
2525import { canAddNewModeledMethod } from "../../model-editor/shared/multiple-modeled-methods" ;
2626import { DataGridCell , DataGridRow } from "../common/DataGrid" ;
2727
28- const MultiModelColumn = styled . div `
28+ const MultiModelColumn = styled ( DataGridCell ) `
2929 display: flex;
3030 flex-direction: column;
3131 gap: 0.5em;
@@ -203,81 +203,71 @@ const ModelableMethodRow = forwardRef<HTMLElement | undefined, MethodRowProps>(
203203 ) }
204204 { ! props . modelingInProgress && (
205205 < >
206- < DataGridCell gridRow = { gridRow } gridColumn = { 2 } >
207- < MultiModelColumn >
208- { modeledMethods . map ( ( modeledMethod , index ) => (
209- < ModelTypeDropdown
210- key = { index }
211- method = { method }
212- modeledMethod = { modeledMethod }
213- onChange = { modeledMethodChangedHandlers [ index ] }
214- />
215- ) ) }
216- </ MultiModelColumn >
217- </ DataGridCell >
218- < DataGridCell gridRow = { gridRow } gridColumn = { 3 } >
219- < MultiModelColumn >
220- { modeledMethods . map ( ( modeledMethod , index ) => (
221- < ModelInputDropdown
222- key = { index }
223- method = { method }
224- modeledMethod = { modeledMethod }
225- onChange = { modeledMethodChangedHandlers [ index ] }
226- />
227- ) ) }
228- </ MultiModelColumn >
229- </ DataGridCell >
230- < DataGridCell gridRow = { gridRow } gridColumn = { 4 } >
231- < MultiModelColumn >
232- { modeledMethods . map ( ( modeledMethod , index ) => (
233- < ModelOutputDropdown
234- key = { index }
235- method = { method }
236- modeledMethod = { modeledMethod }
237- onChange = { modeledMethodChangedHandlers [ index ] }
238- />
239- ) ) }
240- </ MultiModelColumn >
241- </ DataGridCell >
242- < DataGridCell gridRow = { gridRow } gridColumn = { 5 } >
243- < MultiModelColumn >
244- { modeledMethods . map ( ( modeledMethod , index ) => (
245- < ModelKindDropdown
246- key = { index }
247- method = { method }
248- modeledMethod = { modeledMethod }
249- onChange = { modeledMethodChangedHandlers [ index ] }
250- />
251- ) ) }
252- </ MultiModelColumn >
253- </ DataGridCell >
206+ < MultiModelColumn gridRow = { gridRow } gridColumn = { 2 } >
207+ { modeledMethods . map ( ( modeledMethod , index ) => (
208+ < ModelTypeDropdown
209+ key = { index }
210+ method = { method }
211+ modeledMethod = { modeledMethod }
212+ onChange = { modeledMethodChangedHandlers [ index ] }
213+ />
214+ ) ) }
215+ </ MultiModelColumn >
216+ < MultiModelColumn gridRow = { gridRow } gridColumn = { 3 } >
217+ { modeledMethods . map ( ( modeledMethod , index ) => (
218+ < ModelInputDropdown
219+ key = { index }
220+ method = { method }
221+ modeledMethod = { modeledMethod }
222+ onChange = { modeledMethodChangedHandlers [ index ] }
223+ />
224+ ) ) }
225+ </ MultiModelColumn >
226+ < MultiModelColumn gridRow = { gridRow } gridColumn = { 4 } >
227+ { modeledMethods . map ( ( modeledMethod , index ) => (
228+ < ModelOutputDropdown
229+ key = { index }
230+ method = { method }
231+ modeledMethod = { modeledMethod }
232+ onChange = { modeledMethodChangedHandlers [ index ] }
233+ />
234+ ) ) }
235+ </ MultiModelColumn >
236+ < MultiModelColumn gridRow = { gridRow } gridColumn = { 5 } >
237+ { modeledMethods . map ( ( modeledMethod , index ) => (
238+ < ModelKindDropdown
239+ key = { index }
240+ method = { method }
241+ modeledMethod = { modeledMethod }
242+ onChange = { modeledMethodChangedHandlers [ index ] }
243+ />
244+ ) ) }
245+ </ MultiModelColumn >
254246 { viewState . showMultipleModels && (
255- < DataGridCell gridRow = { gridRow } gridColumn = { 6 } >
256- < MultiModelColumn >
257- { modeledMethods . map ( ( _ , index ) =>
258- index === modeledMethods . length - 1 ? (
259- < CodiconRow
260- key = { index }
261- appearance = "icon"
262- aria-label = "Add new model"
263- onClick = { handleAddModelClick }
264- disabled = { addModelButtonDisabled }
265- >
266- < Codicon name = "add" />
267- </ CodiconRow >
268- ) : (
269- < CodiconRow
270- key = { index }
271- appearance = "icon"
272- aria-label = "Remove model"
273- onClick = { removeModelClickedHandlers [ index ] }
274- >
275- < Codicon name = "trash" />
276- </ CodiconRow >
277- ) ,
278- ) }
279- </ MultiModelColumn >
280- </ DataGridCell >
247+ < MultiModelColumn gridRow = { gridRow } gridColumn = { 6 } >
248+ { modeledMethods . map ( ( _ , index ) =>
249+ index === modeledMethods . length - 1 ? (
250+ < CodiconRow
251+ key = { index }
252+ appearance = "icon"
253+ aria-label = "Add new model"
254+ onClick = { handleAddModelClick }
255+ disabled = { addModelButtonDisabled }
256+ >
257+ < Codicon name = "add" />
258+ </ CodiconRow >
259+ ) : (
260+ < CodiconRow
261+ key = { index }
262+ appearance = "icon"
263+ aria-label = "Remove model"
264+ onClick = { removeModelClickedHandlers [ index ] }
265+ >
266+ < Codicon name = "trash" />
267+ </ CodiconRow >
268+ ) ,
269+ ) }
270+ </ MultiModelColumn >
281271 ) }
282272 </ >
283273 ) }
0 commit comments