From ce66cb99cb8495a9f56b3a3ba7a662625f520eef Mon Sep 17 00:00:00 2001
From: Tasha Upchurch <TashaSkyUp@gmail.com>
Date: Tue, 10 Jun 2025 04:40:34 +0000
Subject: [PATCH] fix

---
 pytorch_nodes.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/pytorch_nodes.py b/pytorch_nodes.py
index 9b4e154..ca8c6ec 100644
--- a/pytorch_nodes.py
+++ b/pytorch_nodes.py
@@ -11,10 +11,7 @@ try:
 except ImportError as e:
     from config import config_settings
 
-# from pytorch_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
-# from . import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
-NODE_CLASS_MAPPINGS = {}
-NODE_DISPLAY_NAME_MAPPINGS = {}
+from . import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
 
 
 def ETK_pytorch_base(cls):
@@ -814,3 +811,4 @@ class TrainModel:
                     best_model = torch.load(tmp, weights_only=False)
             # Clean return, no debug prints
             return (model, metrics_out, best_model,)
+