update e2e

Author: paula-stacho

packages/compass-e2e-tests/tests/data-modeling-tab.test.ts

@@ -96,12 +96,14 @@ describe('Data Modeling tab', function () {
     const dataModelEditor = browser.$(Selectors.DataModelEditor);
     await dataModelEditor.waitForDisplayed();
 
-    let nodes = await getDiagramNodes(browser);
-    expect(nodes).to.have.lengthOf(2);
-    expect(nodes).to.deep.equal([
-      'test.testCollection1',
-      'test.testCollection2',
-    ]);
+    browser.waitUntil(async () => {
+      let nodes = await getDiagramNodes(browser);
+      expect(nodes).to.have.lengthOf(2);
+      expect(nodes).to.deep.equal([
+        'test.testCollection1',
+        'test.testCollection2',
+      ]);
+    });
 
     // Apply change to the model
     const newModel = {

packages/compass-web/sandbox/index.tsx

@@ -137,7 +137,7 @@ const App = () => {
                 isAtlas && !!enableGenAIFeaturesAtlasOrg,
               optInDataExplorerGenAIFeatures:
                 isAtlas && !!optInDataExplorerGenAIFeatures,
-              enableDataModeling: false,
+              enableDataModeling: true,
             }}
             onTrack={sandboxTelemetry.track}
             onDebug={sandboxLogger.debug}