From 92cdf19eaec722d1a1e889dddf704145d5bf9a69 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= <cedric@deltheil.me>
Date: Fri, 1 Sep 2023 10:36:23 +0200
Subject: [PATCH] add Distribute to fluxion layers's __init__.py

---
 src/refiners/fluxion/layers/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/refiners/fluxion/layers/__init__.py b/src/refiners/fluxion/layers/__init__.py
index a48b747..4b55c64 100644
--- a/src/refiners/fluxion/layers/__init__.py
+++ b/src/refiners/fluxion/layers/__init__.py
@@ -24,6 +24,7 @@ from refiners.fluxion.layers.chain import (
     UseContext,
     SetContext,
     Parallel,
+    Distribute,
     Passthrough,
     Breakpoint,
     Concatenate,
@@ -71,6 +72,7 @@ __all__ = [
     "UseContext",
     "SetContext",
     "Parallel",
+    "Distribute",
     "Passthrough",
     "Breakpoint",
     "Concatenate",