diff --git a/pytorch_nodes.py b/pytorch_nodes.py
index 3f96d7c..9b4e154 100644
--- a/pytorch_nodes.py
+++ b/pytorch_nodes.py
@@ -11,7 +11,7 @@ try:
 except ImportError as e:
     from config import config_settings
 
-from pytorch_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+# 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 = {}