Skip to content
Snippets Groups Projects
Commit 2178b66e authored by Markus Holzer's avatar Markus Holzer
Browse files

Use Walrus operator

parent a27c182a
No related branches found
No related tags found
No related merge requests found
Pipeline #36486 failed
...@@ -453,10 +453,8 @@ class CustomSympyPrinter(CCodePrinter): ...@@ -453,10 +453,8 @@ class CustomSympyPrinter(CCodePrinter):
def _print_Rational(self, expr): def _print_Rational(self, expr):
"""Evaluate all rationals i.e. print 0.25 instead of 1.0/4.0""" """Evaluate all rationals i.e. print 0.25 instead of 1.0/4.0"""
res = str(expr.evalf(17)) res = str(expr.evalf(17))
if "." in res: if p := res.find('.'):
res = res.rstrip('0') res = res[:p + 2] + res[p + 2:].rstrip('0')
if res[-1] == ".":
res += '0'
return res return res
def _print_Equality(self, expr): def _print_Equality(self, expr):
...@@ -529,10 +527,8 @@ class CustomSympyPrinter(CCodePrinter): ...@@ -529,10 +527,8 @@ class CustomSympyPrinter(CCodePrinter):
def _typed_number(self, number, dtype): def _typed_number(self, number, dtype):
res = self._print(number) res = self._print(number)
if "." in res: if p := res.find('.'):
res = res.rstrip('0') res = res[:p + 2] + res[p + 2:].rstrip('0')
if res[-1] == ".":
res += '0'
if dtype.numpy_dtype == np.float32: if dtype.numpy_dtype == np.float32:
return res + '.0f' if '.' not in res else res + 'f' return res + '.0f' if '.' not in res else res + 'f'
elif dtype.numpy_dtype == np.float64: elif dtype.numpy_dtype == np.float64:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment