fix

2025-06-10 04:40:34 +00:00
parent f092d9a0c1
commit ce66cb99cb
1 changed files with 2 additions and 4 deletions
@@ -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,)
+