fix find stdlib in flat pxr lib

PrismPipeline · Aug 2, 2023 · 5348bf2 · 5348bf2
1 parent e0d959f
commit 5348bf2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/QuiltiX/mx_node.py b/src/QuiltiX/mx_node.py
@@ -58,6 +58,10 @@ def recurse_find_mx_stdlib_in_dir(root_dir):
         return [Path(Path(sys.executable).parent.parent, "houdini", "materialx", "libraries").as_posix()]
 
     import pxr  # type: ignore
+    usd_root = os.path.dirname(pxr.__file__)  # flat pxr module
+    if pxr_stdlib := recurse_find_mx_stdlib_in_dir(usd_root):
+        return [pxr_stdlib]
+
     usd_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(pxr.__file__))))
     if pxr_stdlib := recurse_find_mx_stdlib_in_dir(usd_root):
         return [pxr_stdlib]