@@ -49,7 +49,7 @@ export default function PredictionCreationDialog({
4949 const [ manualRows , setManualRows ] = useState ( [ ] ) ;
5050 const [ isLoading , setIsLoading ] = useState ( false ) ;
5151
52- const [ experiment , setExperiment ] = useState ( null ) ;
52+ const [ modelSession , setModelSession ] = useState ( null ) ;
5353 const [ types , setTypes ] = useState ( { } ) ;
5454 const [ sample , setSample ] = useState ( null ) ;
5555 const [ loadingExperiment , setLoadingExperiment ] = useState ( true ) ;
@@ -83,18 +83,16 @@ export default function PredictionCreationDialog({
8383 ) ;
8484 setDatasets ( availableDatasetsWithInfo ) ;
8585
86- if ( run . experiment_id || session ?. id ) {
87- const experimentData = await getModelSessionById (
88- run . experiment_id || session . id ,
86+ if ( run . model_session_id || session ?. id ) {
87+ const sessionData = await getModelSessionById (
88+ run . model_session_id || session . id ,
8989 ) ;
90- setExperiment ( experimentData ) ;
90+ setModelSession ( sessionData ) ;
9191
92- const datasetTypes = await getDatasetTypes ( experimentData . dataset_id ) ;
92+ const datasetTypes = await getDatasetTypes ( sessionData . dataset_id ) ;
9393 setTypes ( datasetTypes ) ;
9494
95- const datasetSample = await getDatasetSample (
96- experimentData . dataset_id ,
97- ) ;
95+ const datasetSample = await getDatasetSample ( sessionData . dataset_id ) ;
9896 setSample ( datasetSample ) ;
9997 }
10098 } catch ( error ) {
@@ -228,7 +226,7 @@ export default function PredictionCreationDialog({
228226 Select a dataset for prediction
229227 </ Typography >
230228 < DatasetSelector
231- experiment = { experiment }
229+ experiment = { modelSession }
232230 datasets = { datasets }
233231 selectedDataset = { selectedDataset }
234232 setSelectedDataset = { setSelectedDataset }
@@ -244,7 +242,7 @@ export default function PredictionCreationDialog({
244242 Enter manual input data
245243 </ Typography >
246244 < ManualInput
247- experiment = { experiment }
245+ experiment = { modelSession }
248246 loading = { loadingExperiment }
249247 types = { types }
250248 sample = { sample }
@@ -371,7 +369,7 @@ PredictionCreationDialog.propTypes = {
371369 id : PropTypes . number . isRequired ,
372370 name : PropTypes . string ,
373371 model_name : PropTypes . string ,
374- experiment_id : PropTypes . number ,
372+ model_session_id : PropTypes . number ,
375373 } ) . isRequired ,
376374 session : PropTypes . shape ( {
377375 id : PropTypes . number ,
0 commit comments