From 52c522d019699e6c8d7f8c7694fcbe0c1c5498f1 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Tue, 17 Jun 2025 14:46:46 -0700
Subject: [PATCH] nodes train fix, nodes stich test imports removal

---
 {comfy_Extras => comfy_extras}/nodes/nodes_train.py | 2 +-
 tests/unit/comfy_extras_test/image_stitch_test.py   | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)
 rename {comfy_Extras => comfy_extras}/nodes/nodes_train.py (99%)

diff --git a/comfy_Extras/nodes/nodes_train.py b/comfy_extras/nodes/nodes_train.py
similarity index 99%
rename from comfy_Extras/nodes/nodes_train.py
rename to comfy_extras/nodes/nodes_train.py
index a3951ac9b..10a70c559 100644
--- a/comfy_Extras/nodes/nodes_train.py
+++ b/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):
diff --git a/tests/unit/comfy_extras_test/image_stitch_test.py b/tests/unit/comfy_extras_test/image_stitch_test.py
index d845c0a01..eb64516b5 100644
--- a/tests/unit/comfy_extras_test/image_stitch_test.py
+++ b/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