Add missing python files to wheel

Author: njroussel

CMakeLists.txt

@@ -209,15 +209,19 @@ if (DRJIT_MASTER_PROJECT)
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/config.py
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/__init__.pyi
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/__init__.py
-      ${CMAKE_CURRENT_BINARY_DIR}/drjit/interop.pyi
-      ${CMAKE_CURRENT_BINARY_DIR}/drjit/interop.py
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/ast.pyi
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/ast.py
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/dda.py
-      ${CMAKE_CURRENT_BINARY_DIR}/drjit/nn.py
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/detail.pyi
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/detail.py
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/hashgrid.pyi
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/hashgrid.py
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/interop.pyi
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/interop.py
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/nn.pyi
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/nn.py
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/opt.py
+      ${CMAKE_CURRENT_BINARY_DIR}/drjit/random.pyi
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/random.py
       ${CMAKE_CURRENT_BINARY_DIR}/drjit/py.typed
       DESTINATION drjit

drjit/__init__.py

@@ -30,7 +30,7 @@
 
 from .ast import syntax, hint
 from .interop import wrap
-import drjit.random
+from . import random
 import warnings as _warnings
 
 from functools import wraps

src/python/CMakeLists.txt

@@ -196,10 +196,10 @@ if (NOT (DRJIT_SANITIZE_ASAN OR DRJIT_SANITIZE_UBSAN))
     ${DRJIT_PYTHON_DST_DIR}/__init__.pyi
     ${DRJIT_PYTHON_DST_DIR}/ast.pyi
     ${DRJIT_PYTHON_DST_DIR}/interop.pyi
-    ${DRJIT_PYTHON_DST_DIR}/random.pyi
     ${DRJIT_PYTHON_DST_DIR}/detail.pyi
-    ${DRJIT_PYTHON_DST_DIR}/nn.pyi
     ${DRJIT_PYTHON_DST_DIR}/hashgrid.pyi
+    ${DRJIT_PYTHON_DST_DIR}/nn.pyi
+    ${DRJIT_PYTHON_DST_DIR}/random.pyi
     ${DRJIT_PYTHON_DST_DIR}/scalar/__init__.pyi
   )