Add beams to __init__ and make the pspecbeam namespace a bit more acc…

…essible
HERA-Team · Apr 27, 2018 · c6de44e · c6de44e
1 parent 3787d30
commit c6de44e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hera_pspec/__init__.py b/hera_pspec/__init__.py
@@ -7,7 +7,8 @@
 from hera_pspec.pspecdata import PSpecData
 from hera_pspec.container import PSpecContainer
 from hera_pspec.parameter import PSpecParam
-from hera_pspec.pspecbeam import PSpecBeamUV
+from hera_pspec import pspecbeam as beam
+from hera_pspec.pspecbeam import PSpecBeamUV, PSpecBeamGauss, PSpecBeamFromArray
 
 # XXX: This will eventually be deprecated
 from hera_pspec import legacy_pspec as legacy