diff options
-rw-r--r-- | node_wrangler.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/node_wrangler.py b/node_wrangler.py index 290ac082..cc1b85fe 100644 --- a/node_wrangler.py +++ b/node_wrangler.py @@ -830,12 +830,13 @@ class NWNodeWrangler(bpy.types.AddonPreferences): def nw_check(context): space = context.space_data - editor_is_valid = space.type == 'NODE_EDITOR' - valid_trees = ["ShaderNodeTree", "CompositorNodeTree", "TextureNodeTree", "GeometryNodeTree"] - tree_is_valid = space.node_tree is not None and space.tree_type in valid_trees - return editor_is_valid and tree_is_valid + valid = False + if space.type == 'NODE_EDITOR' and space.node_tree is not None and space.tree_type in valid_trees: + valid = True + + return valid class NWBase: @classmethod |