Expose Sampler base class in public API.

PiperOrigin-RevId: 395124580
google-deepmind · Sep 6, 2021 · 762165c · 762165c
1 parent db8f86d
commit 762165c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/clrs/__init__.py b/clrs/__init__.py
@@ -26,6 +26,7 @@
 from clrs._src.samplers import clrs21_val
 from clrs._src.samplers import Features
 from clrs._src.samplers import Feedback
+from clrs._src.samplers import Sampler
 from clrs._src.samplers import Trajectory
 from clrs._src.specs import Location
 from clrs._src.specs import OutputClass
@@ -48,6 +49,7 @@
     "Feedback",
     "Location",
     "Model",
+    "Sampler",
     "Spec",
     "SPECS",
     "Stage",