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)