update stories

Arize-ai · Jun 14, 2023 · b6e33cd · b6e33cd
1 parent 3288b80
commit b6e33cd
Showing 1 changed file with 25 additions and 27 deletions.
diff --git a/stories/LassoSelect.stories.tsx b/stories/LassoSelect.stories.tsx
@@ -87,34 +87,32 @@ function TwoDPointCloudWithSelect(props) {
   }, []);
 
   return (
-    <div style={{ position: 'relative' }}>
-      <TwoDimensionalCanvas camera={{ zoom: 30, up: [0, 0, 1] }}>
-        <TwoDimensionalBounds bounds={bounds}>
-          <TwoDimensionalControls />
-          <LassoSelect
-            points={twoDData}
-            onChange={(selection) => {
-              props.onChange(selection);
-            }}
-            enabled={selectedTool === 'select'}
-          />
-          <axesHelper />
-          <Points
-            data={twoDData}
-            pointProps={{
-              color: (p) => {
-                if (props.selectedPoints.includes(p.metaData.uuid)) {
-                  return '#40E0D0';
-                } else if (props.selectedPoints.length) {
-                  return '#216c64';
-                }
+    <TwoDimensionalCanvas camera={{ zoom: 30, up: [0, 0, 1] }}>
+      <TwoDimensionalBounds bounds={bounds}>
+        <TwoDimensionalControls enabled={selectedTool === 'move'} />
+        <LassoSelect
+          points={twoDData}
+          onChange={(selection) => {
+            props.onChange(selection);
+          }}
+          enabled={selectedTool === 'select'}
+        />
+        <axesHelper />
+        <Points
+          data={twoDData}
+          pointProps={{
+            color: (p) => {
+              if (selectedPoints.includes(p.metaData.uuid)) {
                 return '#40E0D0';
-              },
-            }}
-          />
-        </TwoDimensionalBounds>
-      </TwoDimensionalCanvas>
-    </div>
+              } else if (props.selectedPoints.length) {
+                return '#216c64';
+              }
+              return '#40E0D0';
+            },
+          }}
+        />
+      </TwoDimensionalBounds>
+    </TwoDimensionalCanvas>
   );
 }