From a92a955d1e6a98cd4d334c2a48a81c465455f3b9 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Wed, 13 Sep 2017 18:17:22 +0800
Subject: [PATCH] rtio/sed: use __all__

---
 artiq/gateware/rtio/sed/lane_distributor.py | 3 +++
 artiq/gateware/rtio/sed/output_driver.py    | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/artiq/gateware/rtio/sed/lane_distributor.py b/artiq/gateware/rtio/sed/lane_distributor.py
index ddd19ee33..9b3a17e64 100644
--- a/artiq/gateware/rtio/sed/lane_distributor.py
+++ b/artiq/gateware/rtio/sed/lane_distributor.py
@@ -3,6 +3,9 @@ from migen import *
 from artiq.gateware.rtio import cri
 
 
+__all__ = ["LaneDistributor"]
+
+
 def layout_lane_io(seqn_width, layout_payload):
     return [
         ("we", 1, DIR_M_TO_S),
diff --git a/artiq/gateware/rtio/sed/output_driver.py b/artiq/gateware/rtio/sed/output_driver.py
index ad9a6c2e9..5ee24753e 100644
--- a/artiq/gateware/rtio/sed/output_driver.py
+++ b/artiq/gateware/rtio/sed/output_driver.py
@@ -7,6 +7,9 @@ from artiq.gateware.rtio import rtlink
 from artiq.gateware.rtio.sed.output_network import OutputNetwork
 
 
+__all__ = ["OutputDriver"]
+
+
 class OutputDriver(Module):
     def __init__(self, channels, lane_count, seqn_width):
         self.collision = Signal()