From b4fee6e7af52d2f59f77db0a023b1b0b45961f1e Mon Sep 17 00:00:00 2001 From: George Bisbas Date: Tue, 16 Jul 2024 08:54:20 +0100 Subject: [PATCH] flake8 --- devito/ir/xdsl_iet/cluster_to_ssa.py | 8 ++++---- tests/test_xdsl_op_correctness.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/devito/ir/xdsl_iet/cluster_to_ssa.py b/devito/ir/xdsl_iet/cluster_to_ssa.py index 4f95c9d678..787b5f38cf 100644 --- a/devito/ir/xdsl_iet/cluster_to_ssa.py +++ b/devito/ir/xdsl_iet/cluster_to_ssa.py @@ -296,11 +296,11 @@ def _visit_math_nodes(self, dim: SteppingDimension, node: Expr, elif isinstance(node, cos): assert len(node.args) == 1, "Expected single argument for cos." - # import pdb; pdb.set_trace() + return math.CosOp(self._visit_math_nodes(dim, node.args[0], output_indexed)).result - # import pdb; pdb.set_trace() + # return reduce(lambda x, y : arith.AndI(x, y).result, SSAargs) elif isinstance(node, Relational): @@ -515,7 +515,7 @@ def build_time_loop( def lower_devito_Eqs(self, eqs: list[Any], **kwargs): # Lower devito Equations to xDSL - # import pdb; pdb.set_trace() + for eq in eqs: lowered = self.operator._lower_exprs(as_tuple(eq), **kwargs) if isinstance(eq, Eq): @@ -648,7 +648,7 @@ def convert(self, eqs: Iterable[Eq], **kwargs) -> ModuleOp: functions.add(f.function) elif isinstance(eq, Injection): - # import pdb; pdb.set_trace() + functions.add(eq.field.function) for f in retrieve_functions(eq.expr): if isinstance(f, PointSource): diff --git a/tests/test_xdsl_op_correctness.py b/tests/test_xdsl_op_correctness.py index f04770bd9f..e0356d5fde 100644 --- a/tests/test_xdsl_op_correctness.py +++ b/tests/test_xdsl_op_correctness.py @@ -187,7 +187,7 @@ def test_extraction_from_lifted_ispace(rotate): @pytest.mark.parametrize('shape', [(101, 101), (201, 201)]) @pytest.mark.parametrize('nt', [10, 20, ]) -def test_dt2_script(shape, nt): +def test_dt2_sources_script(shape, nt): spacing = (10., 10.) # spacing of 10 meters nbl = 1 # number of pad layers