Add docstring

eigenvivek · Nov 30, 2023 · 51ceafe · 51ceafe
1 parent d39cb5d
commit 51ceafe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/diffpose/calibration.py b/diffpose/calibration.py
@@ -27,6 +27,7 @@ def __init__(
         device=None,
         dtype=torch.float32,
     ):
+        """Wrapper of pytorch3d.transforms.Transform3d with extra functionalities."""
         if device is None and (R.device == t.device):
             device = R.device
 

diff --git a/notebooks/api/02_calibration.ipynb b/notebooks/api/02_calibration.ipynb
@@ -114,6 +114,7 @@
     "        device=None,\n",
     "        dtype=torch.float32,\n",
     "    ):\n",
+    "        \"\"\"Wrapper of pytorch3d.transforms.Transform3d with extra functionalities.\"\"\"\n",
     "        if device is None and (R.device == t.device):\n",
     "            device = R.device\n",
     "\n",