nodes train fix, nodes stich test imports removal

comfy_extras/nodes/nodes_train.py

@@ -20,7 +20,7 @@ from comfy import node_helpers
 from comfy.cli_args import args
 from comfy.comfy_types.node_typing import IO
 from comfy.weight_adapter import adapters
-from .nodes import nodes_custom_sampler
+from . import nodes_custom_sampler
 
 
 class TrainSampler(comfy.samplers.Sampler):

tests/unit/comfy_extras_test/image_stitch_test.py

@@ -1,6 +1,5 @@
 import pytest
 import torch
-from unittest.mock import MagicMock
 
 from comfy_extras.nodes.nodes_images import ImageStitch