[https://nvbugs/5628848][fix] Fix nanobind stub generation

Signed-off-by: Linda-Stadter <57756729+Linda-Stadter@users.noreply.github.com>

Author: Linda-Stadter

scripts/build_wheel.py

@@ -395,8 +395,13 @@ def generate_python_stubs_linux(binding_type: str, venv_python: Path,
     try:
         if is_nanobind:
             build_run(
-                f"\"{venv_python}\" -m nanobind.stubgen -m bindings -r -O .",
-                env=env_stub_gen)
+                f'"{venv_python}" -m nanobind.stubgen -m bindings -r -O bindings',
+                env=env_stub_gen,
+            )
+            # Root stub needs renaming from bindings.pyi to __init__.pyi
+            bindings_stub = Path("bindings/bindings.pyi")
+            if bindings_stub.exists():
+                bindings_stub.rename("bindings/__init__.pyi")
         else:
             build_run(
                 f"\"{venv_python}\" -m pybind11_stubgen -o . bindings --exit-code",

setup.py

@@ -128,7 +128,6 @@ def has_ext_modules(self):
     ]
 
 package_data += [
-    'bindings.pyi',
     'bindings/*.pyi',
     'tools/plugin_gen/templates/*',
     'bench/build/benchmark_config.yml',