MORE LINTER FIXES

src/more_math/Parser/MathExprVisitor.py

@@ -1,5 +1,5 @@
 # Generated from src/more_math/Parser/MathExpr.g4 by ANTLR 4.13.2
-from antlr4 import *
+from antlr4 import ParseTreeVisitor
 if "." in __name__:
     from .MathExprParser import MathExprParser
 else:

src/more_math/Parser/TensorEvalVisitor.py

@@ -1,9 +1,7 @@
-import re
-from tkinter import SE
 import torch
 import torch.special
 
-from ..helper_functions import freq_to_time, getIndexTensorAlongDim, time_to_freq
+from ..helper_functions import freq_to_time, time_to_freq
 from .MathExprVisitor import MathExprVisitor
 
 class TensorEvalVisitor(MathExprVisitor):

src/more_math/VideoMathNode.py

@@ -1,8 +1,8 @@
 from inspect import cleandoc
 
-from comfy_api.latest import ComfyExtension, io
-from comfy_api.input_impl import VideoFromComponents, VideoFromFile
-from comfy_api.util import VideoCodec, VideoComponents, VideoContainer
+from comfy_api.latest import io
+from comfy_api.input_impl import VideoFromComponents
+from comfy_api.util import VideoComponents
 
 
 from antlr4 import CommonTokenStream, InputStream

tests/test_more_math.py

@@ -356,42 +356,3 @@ def test_nested_tensor_support():
     assert torch.allclose(res_list[1], torch.full_like(t2, 3.0))
 #!/usr/bin/env python
 
-
-# ==========================================
-# Node Initialization and Metadata Tests
-# ==========================================
-
-def test_conditioning_math_node_initialization():
-    node = ConditioningMathNode()
-    assert isinstance(node, ConditioningMathNode)
-
-
-def test_conditioning_math_node_metadata():
-    assert ConditioningMathNode.RETURN_TYPES == ["CONDITIONING"]
-    assert ConditioningMathNode.FUNCTION == "EXECUTE_NORMALIZED"
-    assert ConditioningMathNode.CATEGORY == "More math"
-
-
-def test_latent_math_node_initialization():
-    node = LatentMathNode()
-    assert isinstance(node, LatentMathNode)
-
-
-def test_latent_math_node_metadata():
-    assert LatentMathNode.RETURN_TYPES == ["LATENT"]
-    assert LatentMathNode.FUNCTION == "EXECUTE_NORMALIZED"
-    assert LatentMathNode.CATEGORY == "More math"
-
-
-def test_image_math_node_initialization():
-    node = ImageMathNode()
-    assert isinstance(node, ImageMathNode)
-
-
-def test_image_math_node_metadata():
-    assert ImageMathNode.RETURN_TYPES == ["IMAGE"]
-    assert ImageMathNode.FUNCTION == "EXECUTE_NORMALIZED"
-    assert ImageMathNode.CATEGORY == "More math"
-
-
-# (rest of file unchanged...)