Commit 7cf34d58 authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files

Fix syntax error with python 2 in covariate_blockmodel.py

This fixes #215.
parent 76dbca5b
......@@ -266,12 +266,12 @@ class CovariateBlockState(BlockState):
def __merge_decorator(func):
@contextlib.contextmanager
def context_wrapper(*args, revert=False):
return func(*args, revert=revert)
def context_wrapper(self, l_src, l_tgt, revert=False):
return func(self, l_src, l_tgt, revert=revert)
@_wraps(func)
def wrapper(*args, revert=False):
gen = context_wrapper(*args, revert=revert)
def wrapper(self, l_src, l_tgt, revert=False):
gen = context_wrapper(self, l_src, l_tgt, revert=revert)
if revert:
return gen
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment