diff --git a/openmetadata-ui/src/main/resources/ui/cypress/e2e/AddNewService/MlFlow.spec.js b/openmetadata-ui/src/main/resources/ui/cypress/e2e/AddNewService/MlFlow.spec.js index 8cd09111fe8..3a236e0f814 100644 --- a/openmetadata-ui/src/main/resources/ui/cypress/e2e/AddNewService/MlFlow.spec.js +++ b/openmetadata-ui/src/main/resources/ui/cypress/e2e/AddNewService/MlFlow.spec.js @@ -26,8 +26,8 @@ import { const serviceType = 'Mlflow'; const serviceName = `${serviceType}-ct-test-${uuid()}`; -const tableName = 'ElasticnetWineModel'; -const description = `This is ${tableName} description`; +const modelName = 'ElasticnetWineModel'; +const description = `This is ${modelName} description`; const connectionInput = () => { cy.get('#root\\/trackingUri').type(Cypress.env('mlModelTrackingUri')); @@ -36,6 +36,12 @@ const connectionInput = () => { checkServiceFieldSectionHighlighting('registryUri'); }; +const addIngestionInput = () => { + cy.get('#root\\/mlModelFilterPattern\\/includes') + .scrollIntoView() + .type(`${modelName}{enter}`); +}; + describe('ML Flow Ingestion', () => { beforeEach(() => { cy.login(); @@ -47,6 +53,7 @@ describe('ML Flow Ingestion', () => { testServiceCreationAndIngestion({ serviceType, connectionInput, + addIngestionInput, serviceName, type: SERVICE_TYPE.MLModels, serviceCategory: 'MlModel', @@ -56,7 +63,7 @@ describe('ML Flow Ingestion', () => { it('Update MlModel description and verify description after re-run', () => { updateDescriptionForIngestedTables( serviceName, - tableName, + modelName, description, SERVICE_TYPE.MLModels, MYDATA_SUMMARY_OPTIONS.mlmodels