diff --git a/benchmarks/bench_translations.py b/benchmarks/bench_translations.py index 6e87370f2da1d2a568365c088cdd8ee047b3c8c6..8a9ef0da1dde366ebd6b0d681029c56ae2105093 100644 --- a/benchmarks/bench_translations.py +++ b/benchmarks/bench_translations.py @@ -66,8 +66,14 @@ class TranslationBenchmarkSuite: src_rscale = sym.Symbol("src_rscale") tgt_rscale = sym.Symbol("tgt_rscale") sac = SymbolicAssignmentCollection() - result = l_expn.translate_from(m_expn, src_coeff_exprs, src_rscale, + try: + result = l_expn.translate_from(m_expn, src_coeff_exprs, src_rscale, dvec, tgt_rscale, sac) + except TypeError: + # Support older interface to make it possible to compare + # in CI run + result = l_expn.translate_from(m_expn, src_coeff_exprs, src_rscale, + dvec, tgt_rscale) for i, expr in enumerate(result): sac.assign_unique("coeff%d" % i, expr) sac.run_global_cse()