From 9977858e28c603cc7bedfe6a597e59272a5ab4da Mon Sep 17 00:00:00 2001
From: Neil Kistner <neil.kistner@gmail.com>
Date: Thu, 17 Jun 2021 22:14:51 -0600
Subject: [PATCH] fix(swarmplot): pass renderWrapper prop to container

---
 packages/swarmplot/src/SwarmPlot.tsx       | 8 ++------
 packages/swarmplot/src/SwarmPlotCanvas.tsx | 8 ++------
 packages/swarmplot/src/types.ts            | 1 +
 3 files changed, 5 insertions(+), 12 deletions(-)

diff --git a/packages/swarmplot/src/SwarmPlot.tsx b/packages/swarmplot/src/SwarmPlot.tsx
index c66413b38..76e18602f 100644
--- a/packages/swarmplot/src/SwarmPlot.tsx
+++ b/packages/swarmplot/src/SwarmPlot.tsx
@@ -212,15 +212,11 @@ export const SwarmPlot = <RawDatum,>({
     isInteractive = defaultProps.isInteractive,
     animate = defaultProps.animate,
     motionConfig = defaultProps.motionConfig,
+    renderWrapper,
     ...otherProps
 }: Partial<Omit<SwarmPlotSvgProps<RawDatum>, 'data' | 'groups' | 'width' | 'height'>> &
     Pick<SwarmPlotSvgProps<RawDatum>, 'data' | 'groups' | 'width' | 'height'>) => (
-    <Container
-        isInteractive={isInteractive}
-        animate={animate}
-        motionConfig={motionConfig}
-        theme={theme}
-    >
+    <Container {...{ isInteractive, animate, motionConfig, theme, renderWrapper }}>
         <InnerSwarmPlot<RawDatum> isInteractive={isInteractive} {...otherProps} />
     </Container>
 )
diff --git a/packages/swarmplot/src/SwarmPlotCanvas.tsx b/packages/swarmplot/src/SwarmPlotCanvas.tsx
index 27143fa92..f8dcc86cb 100644
--- a/packages/swarmplot/src/SwarmPlotCanvas.tsx
+++ b/packages/swarmplot/src/SwarmPlotCanvas.tsx
@@ -309,15 +309,11 @@ export const SwarmPlotCanvas = <RawDatum,>({
     isInteractive = defaultProps.isInteractive,
     animate = defaultProps.animate,
     motionConfig = defaultProps.motionConfig,
+    renderWrapper,
     ...otherProps
 }: Partial<Omit<SwarmPlotCanvasProps<RawDatum>, 'data' | 'groups' | 'width' | 'height'>> &
     Pick<SwarmPlotCanvasProps<RawDatum>, 'data' | 'groups' | 'width' | 'height'>) => (
-    <Container
-        isInteractive={isInteractive}
-        animate={animate}
-        motionConfig={motionConfig}
-        theme={theme}
-    >
+    <Container {...{ isInteractive, animate, motionConfig, theme, renderWrapper }}>
         <InnerSwarmPlotCanvas<RawDatum> isInteractive={isInteractive} {...otherProps} />
     </Container>
 )
diff --git a/packages/swarmplot/src/types.ts b/packages/swarmplot/src/types.ts
index 99e95f615..0a2f9952a 100644
--- a/packages/swarmplot/src/types.ts
+++ b/packages/swarmplot/src/types.ts
@@ -129,6 +129,7 @@ export type SwarmPlotCommonProps<RawDatum> = {
     animate: boolean
     motionConfig: ModernMotionProps['motionConfig']
     role: string
+    renderWrapper?: boolean
 }
 
 export type SwarmPlotSvgProps<RawDatum> = SwarmPlotCommonProps<RawDatum> &