You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.4 KiB
TypeScript
59 lines
1.4 KiB
TypeScript
// Copyright (C) 2020 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
import { connect } from 'react-redux';
|
|
|
|
import ModelRunnerModalComponent from 'components/model-runner-modal/model-runner-modal';
|
|
import { Model, CombinedState } from 'reducers/interfaces';
|
|
import { startInferenceAsync, modelsActions } from 'actions/models-actions';
|
|
|
|
interface StateToProps {
|
|
models: Model[];
|
|
activeProcesses: {
|
|
[index: string]: string;
|
|
};
|
|
taskInstance: any;
|
|
visible: boolean;
|
|
}
|
|
|
|
interface DispatchToProps {
|
|
runInference(
|
|
taskInstance: any,
|
|
model: Model,
|
|
body: object,
|
|
): void;
|
|
closeDialog(): void;
|
|
}
|
|
|
|
function mapStateToProps(state: CombinedState): StateToProps {
|
|
const { models } = state;
|
|
|
|
return {
|
|
models: models.models,
|
|
activeProcesses: {},
|
|
taskInstance: models.activeRunTask,
|
|
visible: models.visibleRunWindows,
|
|
};
|
|
}
|
|
|
|
function mapDispatchToProps(dispatch: any): DispatchToProps {
|
|
return ({
|
|
runInference(
|
|
taskInstance: any,
|
|
model: Model,
|
|
body: object,
|
|
): void {
|
|
dispatch(startInferenceAsync(taskInstance, model, body));
|
|
},
|
|
closeDialog(): void {
|
|
dispatch(modelsActions.closeRunModelDialog());
|
|
},
|
|
});
|
|
}
|
|
|
|
export default connect(
|
|
mapStateToProps,
|
|
mapDispatchToProps,
|
|
)(ModelRunnerModalComponent);
|