update stubs

Author: fbxiang

python/py_package/__init__.pyi

@@ -1,7 +1,7 @@
 from __future__ import annotations
 from _warnings import warn
 import os as os
-from pathlib import Path
+from pathlib._local import Path
 import pkg_resources as pkg_resources
 import platform as platform
 from sapien.pysapien import Component
@@ -14,10 +14,7 @@ from sapien.pysapien import math
 from sapien.pysapien.physx import PhysxSceneConfig as SceneConfig
 from sapien.pysapien import profile
 from sapien.pysapien import set_log_level
-
-import platform
-if platform.system() != "Darwin":
-    from sapien.pysapien import simsense
+from sapien.pysapien import simsense
 from sapien.pysapien_pinocchio import PinocchioModel
 from sapien.wrapper.actor_builder import ActorBuilder
 from sapien.wrapper.articulation_builder import ArticulationBuilder
@@ -36,10 +33,6 @@ from . import render
 from . import utils
 from . import version
 from . import wrapper
-
-if platform.system() == "Darwin":
-    __all__ = ['ActorBuilder', 'ArticulationBuilder', 'Component', 'CudaArray', 'Device', 'Engine', 'Entity', 'Path', 'PinocchioModel', 'Pose', 'SapienRenderer', 'Scene', 'SceneConfig', 'System', 'Widget', 'asset', 'internal_renderer', 'math', 'os', 'physx', 'pkg_resources', 'platform', 'profile', 'pysapien', 'pysapien_pinocchio', 'render', 'set_log_level', 'utils', 'version', 'warn', 'wrapper']
-else:
-    __all__ = ['ActorBuilder', 'ArticulationBuilder', 'Component', 'CudaArray', 'Device', 'Engine', 'Entity', 'Path', 'PinocchioModel', 'Pose', 'SapienRenderer', 'Scene', 'SceneConfig', 'System', 'Widget', 'asset', 'internal_renderer', 'math', 'os', 'physx', 'pkg_resources', 'platform', 'profile', 'pysapien', 'pysapien_pinocchio', 'render', 'set_log_level', 'simsense', 'utils', 'version', 'warn', 'wrapper']
-__version__: str = '3.0.0.dev20240521+6b6d61d2'
-__warningregistry__: dict = {'version': 0}
+__all__ = ['ActorBuilder', 'ArticulationBuilder', 'Component', 'CudaArray', 'Device', 'Engine', 'Entity', 'Path', 'PinocchioModel', 'Pose', 'SapienRenderer', 'Scene', 'SceneConfig', 'System', 'Widget', 'asset', 'internal_renderer', 'math', 'os', 'physx', 'pkg_resources', 'platform', 'profile', 'pysapien', 'pysapien_pinocchio', 'render', 'set_log_level', 'simsense', 'utils', 'version', 'warn', 'wrapper']
+__version__: str = '3.0.0.dev20250717+93d5fedf'
+__warningregistry__: dict = {'version': 0, ('pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html. The pkg_resources package is slated for removal as early as 2025-11-30. Refrain from using this package or pin to Setuptools<81.', UserWarning, 18): True}

python/py_package/pysapien/__init__.pyi

@@ -1,24 +1,23 @@
 from __future__ import annotations
 import jax
 import numpy
+import numpy.typing
 import torch
 import typing
 from . import internal_renderer
 from . import math
 from . import physx
 from . import render
-
-import platform
-if platform.system() == "Darwin":
-    __all__ = ['Component', 'CudaArray', 'Device', 'Entity', 'Pose', 'Profiler', 'Scene', 'System', 'abi_version', 'compiled_with_cxx11_abi', 'internal_renderer', 'math', 'physx', 'profile', 'pybind11_internals_id', 'pybind11_use_smart_holder', 'render', 'set_log_level']
-else:
-    from . import simsense
-    __all__ = ['Component', 'CudaArray', 'Device', 'Entity', 'Pose', 'Profiler', 'Scene', 'System', 'abi_version', 'compiled_with_cxx11_abi', 'internal_renderer', 'math', 'physx', 'profile', 'pybind11_internals_id', 'pybind11_use_smart_holder', 'render', 'set_log_level', 'simsense']
+from . import simsense
+__all__ = ['Component', 'CudaArray', 'Device', 'Entity', 'Pose', 'Profiler', 'Scene', 'System', 'abi_version', 'compiled_with_cxx11_abi', 'internal_renderer', 'math', 'physx', 'profile', 'pybind11_internals_id', 'pybind11_use_smart_holder', 'render', 'set_log_level', 'simsense']
 _T = typing.TypeVar("_T", Component)
 class Component:
     entity_pose: Pose
     name: str
     pose: Pose
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __init__(self) -> None:
         ...
     def disable(self) -> None:
@@ -50,6 +49,9 @@ class Component:
     def is_enabled(self) -> bool:
         ...
 class CudaArray:
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __init__(self, data: typing.Any) -> None:
         ...
     def dlpack(self) -> typing.Any:
@@ -77,6 +79,9 @@ class CudaArray:
     def typestr(self) -> str:
         ...
 class Device:
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __init__(self, alias: str) -> None:
         ...
     def __repr__(self) -> str:
@@ -103,6 +108,9 @@ class Device:
 class Entity:
     name: str
     pose: Pose
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __init__(self) -> None:
         ...
     def add_component(self, component: Component) -> Entity:
@@ -147,13 +155,16 @@ class Pose:
     p: numpy.ndarray[typing.Literal[3], numpy.dtype[numpy.float32]]
     q: numpy.ndarray[typing.Literal[4], numpy.dtype[numpy.float32]]
     rpy: numpy.ndarray[typing.Literal[3], numpy.dtype[numpy.float32]]
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __getstate__(self) -> tuple:
         ...
     @typing.overload
     def __init__(self, p: numpy.ndarray[typing.Literal[3], numpy.dtype[numpy.float32]] | list[float] | tuple = ..., q: numpy.ndarray[typing.Literal[4], numpy.dtype[numpy.float32]] | list[float] | tuple = ...) -> None:
         ...
     @typing.overload
-    def __init__(self, matrix: numpy.ndarray[tuple[typing.Literal[4], typing.Literal[4]], numpy.dtype[numpy.float32]] | list | tuple) -> None:
+    def __init__(self, matrix: typing.Annotated[numpy.typing.ArrayLike, numpy.float32, "[4, 4]"]) -> None:
         ...
     def __mul__(self, other: Pose) -> Pose:
         ...
@@ -175,9 +186,12 @@ class Pose:
         ...
     def set_rpy(self, rpy: numpy.ndarray[typing.Literal[3], numpy.dtype[numpy.float32]] | list[float] | tuple) -> None:
         ...
-    def to_transformation_matrix(self) -> numpy.ndarray[tuple[typing.Literal[4], typing.Literal[4]], numpy.dtype[numpy.float32]]:
+    def to_transformation_matrix(self) -> typing.Annotated[numpy.typing.NDArray[numpy.float32], "[4, 4]"]:
         ...
 class Profiler:
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __call__(self, func: typing.Callable) -> typing.Callable:
         ...
     def __enter__(self) -> None:
@@ -187,6 +201,9 @@ class Profiler:
     def __init__(self, name: str) -> None:
         ...
 class Scene:
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __init__(self, systems: list[System]) -> None:
         ...
     def add_entity(self, entity: Entity) -> None:
@@ -224,6 +241,9 @@ class Scene:
     def render_system(self) -> render.RenderSystem:
         ...
 class System:
+    @staticmethod
+    def _pybind11_conduit_v1_(*args, **kwargs):
+        ...
     def __init__(self) -> None:
         ...
     def step(self) -> None: