From 7499a7004ef4db785d0cd05cf292fdeff65ca90d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= Date: Wed, 23 Aug 2023 09:43:41 +0200 Subject: [PATCH] split chain with parentheses for and node --- tinygrad/shape/symbolic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/shape/symbolic.py b/tinygrad/shape/symbolic.py index d42e70611d..1c4b781682 100644 --- a/tinygrad/shape/symbolic.py +++ b/tinygrad/shape/symbolic.py @@ -287,5 +287,5 @@ render_python: Dict[Type, Callable] = { ModNode: lambda self,ops,ctx: f"({self.a.render(ops,ctx)}%{self.b})", LtNode: lambda self,ops,ctx: f"({self.a.render(ops,ctx)}<{sym_render(self.b,ops,ctx)})", SumNode: lambda self,ops,ctx: f"({'+'.join(sorted([x.render(ops,ctx) for x in self.nodes]))})", - AndNode: lambda self,ops,ctx: f"({' and '.join(sorted([x.render(ops,ctx) for x in self.nodes]))})" + AndNode: lambda self,ops,ctx: ' and '.join(sorted([f"({x.render(ops, ctx)}" for x in self.nodes])) + len(self.nodes) * ')' } \ No newline at end of file