docs: fix measurement-plot story (#661)

zakodium-oss · Feb 12, 2024 · 48ae20c · 48ae20c
1 parent 3a3bdc6
commit 48ae20c
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/stories/app/measurements/measurement-plot.stories.tsx b/stories/app/measurements/measurement-plot.stories.tsx
@@ -11,8 +11,22 @@ export default {
   title: 'Layout/MeasurementPlot',
 };
 
-export function Control(props: Omit<MeasurementPlotProps, 'measurement'>) {
-  return <MeasurementPlot measurement={irMeasurement} {...props} />;
+export function Control(
+  props: Omit<MeasurementPlotProps, 'measurement' | 'measurementDisplay'> & {
+    color: string;
+  },
+) {
+  const { color, ...rest } = props;
+  return (
+    <MeasurementPlot
+      measurement={irMeasurement}
+      measurementDisplay={{
+        color: { kind: 'fixed', color },
+        visible: true,
+      }}
+      {...rest}
+    />
+  );
 }
 
 Control.args = {
@@ -27,6 +41,7 @@ Control.args = {
   showHorizontalGrid: true,
   showVerticalGrid: true,
   flipHorizontalAxis: false,
+  color: 'red',
 };
 
 Control.argTypes = {