Use DeprecationWarning

nerfstudio-project · Jun 1, 2024 · 072a121 · 072a121
1 parent f0820fd
commit 072a121
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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)