Update drop_dims, now works with more than 2 dims passed

[linpy.git] / pypol / __init__.py

diff --git a/pypol/__init__.py b/pypol/__init__.py
index 7a195ca..fc70bea 100644 (file)
--- a/pypol/__init__.py
+++ b/pypol/__init__.py
@@ -2,13 +2,13 @@
 A polyhedral library based on ISL.
 """
 
-from .linear import Constant, Symbol, symbols
-from .linear import eq, le, lt, ge, gt
-from .linear import Empty, Universe
+from .linexprs import Expression, Constant, Symbol, symbols
+from .polyhedra import Polyhedron, Eq, Ne, Le, Lt, Ge, Gt, Ne, Empty, Universe
+from .domains import Domain, And, Or, Not
 
 
 __all__ = [
-    'Constant', 'Symbol', 'symbols',
-    'eq', 'le', 'lt', 'ge', 'gt',
-    'Empty', 'Universe'
+    'Expression', 'Constant', 'Symbol', 'symbols',
+    'Polyhedron', 'Eq', 'Ne', 'Le', 'Lt', 'Ge', 'Gt', 'Empty', 'Universe',
+    'Domain', 'And', 'Or', 'Not',
 ]