diff options
Diffstat (limited to 'boltzgen/utility')
-rw-r--r-- | boltzgen/utility/__init__.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/boltzgen/utility/__init__.py b/boltzgen/utility/__init__.py index 8927868..fa9c760 100644 --- a/boltzgen/utility/__init__.py +++ b/boltzgen/utility/__init__.py @@ -1,2 +1,7 @@ from . import optimizations from . import ndindex + +from sympy.codegen.ast import Assignment + +def assign(names, definitions): + return list(map(lambda x: Assignment(*x), zip(names, definitions))) |