From 072a1213cbc2c7a2c45a2ec7e3d1ca3a9b242402 Mon Sep 17 00:00:00 2001 From: Brent Yi Date: Fri, 31 May 2024 20:30:05 -0700 Subject: [PATCH] Use DeprecationWarning --- src/viser/_gui_handles.py | 3 +-- src/viser/_viser.py | 2 ++ 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/viser/_gui_handles.py b/src/viser/_gui_handles.py index f39ee1a2d..7a458ad36 100644 --- a/src/viser/_gui_handles.py +++ b/src/viser/_gui_handles.py @@ -42,8 +42,7 @@ class GuiContainerProtocol(Protocol): class SupportsRemoveProtocol(Protocol): - def remove(self) -> None: - ... + def remove(self) -> None: ... @dataclasses.dataclass diff --git a/src/viser/_viser.py b/src/viser/_viser.py index 162bfeca9..8fd66ed0e 100644 --- a/src/viser/_viser.py +++ b/src/viser/_viser.py @@ -42,6 +42,7 @@ def __getattr__(self, name: str) -> Any: if hasattr(self.scene, fixed_name): warnings.warn( f"{type(self).__name__}.{name} has been deprecated, use {type(self).__name__}.scene.{fixed_name} instead. Alternatively, pin to `viser<0.2.0`.", + category=DeprecationWarning, stacklevel=2, ) return object.__getattribute__(self.scene, fixed_name) @@ -50,6 +51,7 @@ def __getattr__(self, name: str) -> Any: if hasattr(self.gui, fixed_name): warnings.warn( f"{type(self).__name__}.{name} has been deprecated, use {type(self).__name__}.gui.{fixed_name} instead. Alternatively, pin to `viser<0.2.0`.", + category=DeprecationWarning, stacklevel=2, ) return object.__getattribute__(self.gui, fixed_name)