From 6005b4208aaab9ad615ab601d642152c36b3003d Mon Sep 17 00:00:00 2001 From: blepping Date: Mon, 19 May 2025 02:29:14 -0600 Subject: [PATCH] Make it possible to comment out lines with hash in expressions --- py/expression/expression.py | 5 +++++ py/expression_handlers.py | 1 - 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/py/expression/expression.py b/py/expression/expression.py index e5c96be..d4ff4ae 100644 --- a/py/expression/expression.py +++ b/py/expression/expression.py @@ -97,6 +97,11 @@ class Expression: @classmethod def tokenize(cls, s): + s = "\n".join( + line.rstrip("\r") + for line in s.split("\n") + if not line.lstrip().startswith("#") + ) yield from (cls.fixup_token(m.group(1)) for m in cls.EXPR_RE.finditer(s)) diff --git a/py/expression_handlers.py b/py/expression_handlers.py index 6b38388..1ed9541 100644 --- a/py/expression_handlers.py +++ b/py/expression_handlers.py @@ -159,7 +159,6 @@ class NewLikeHandler(NormHandler): def handle(self, obj, getter): tensor, values = self.safe_get_all(obj, getter) - print(f"\nNEW TENSOR: {values}") return torch.tensor(values, dtype=tensor.dtype, device=tensor.device)