fix output type surfacemesh in shapenet and shrec16 (NVIDIAGameWorks#759

) Signed-off-by: Clement Fuji Tsang <[email protected]>
muses0229 · Sep 22, 2023 · 3f05196 · 3f05196
1 parent 733dee7
commit 3f05196
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/kaolin/io/shapenet.py b/kaolin/io/shapenet.py
@@ -104,15 +104,15 @@ class ShapeNetV1(Dataset):
 
         * if output_dict=True: a dictionary with the following key-value pairs:
 
-            * 'mesh': containing a namedtuple returned by :func:`kaolin.io.off.import_mesh`.
+            * 'mesh': containing a :class:`kaolin.rep.SurfaceMesh` returned by :func:`kaolin.io.obj.import_mesh`.
             * 'name': the model name (i.e the subfolder name)
             * 'path': the full path to the .off
             * 'synset': the synset associated to the category
             * 'labels': the labels associated to the category (see ``synset_to_labels``)
 
         * if output_dict=False (deprecated): a :class:`KaolinDatasetItem` with the fields:
 
-            * ``data``: containing a namedtuple returned by :func:`kaolin.io.off.import_mesh`.
+            * ``data``: containing a :class:`kaolin.rep.SurfaceMesh` returned by :func:`kaolin.io.obj.import_mesh`.
             * ``attributes``: containing a dictionary with the following key-value pairs:
 
                 * 'name': the model name (i.e the subfolder name)
@@ -292,15 +292,15 @@ class ShapeNetV2(Dataset):
 
         * if output_dict=True: a dictionary with the following key-value pairs:
 
-            * 'mesh': containing a namedtuple returned by :func:`kaolin.io.off.import_mesh`.
+            * 'mesh': containing a :class:`kaolin.rep.SurfaceMesh` returned by :func:`kaolin.io.obj.import_mesh`.
             * 'name': the model name (i.e the subfolder name)
             * 'path': the full path to the .off
             * 'synset': the synset associated to the category
             * 'labels': the labels associated to the category (see ``synset_to_labels``)
 
         * if output_dict=False (deprecated): a :class:`KaolinDatasetItem` with the fields:
 
-            * ``data``: containing a namedtuple returned by :func:`kaolin.io.off.import_mesh`.
+            * ``data``: containing a :class:`kaolin.rep.SurfaceMesh` returned by :func:`kaolin.io.obj.import_mesh`.
             * ``attributes``: containing a dictionary with the following key-value pairs:
 
                 * 'name': the model name (i.e the subfolder name)

diff --git a/kaolin/io/shrec.py b/kaolin/io/shrec.py
@@ -104,15 +104,15 @@ class SHREC16(Dataset):
 
         * if output_dict=True: a dictionary with the following key-value pairs:
 
-            * 'mesh': containing a namedtuple returned by :func:`kaolin.io.off.import_mesh`.
+            * 'mesh': containing a :class:`kaolin.rep.SurfaceMesh` returned by :func:`kaolin.io.obj.import_mesh`.
             * 'name': the model name (i.e the subfolder name)
             * 'path': the full path to the .off
             * 'synset': the synset associated to the category
             * 'labels': the labels associated to the category (see ``synset_to_labels``)
 
         * if output_dict=False (deprecated): a :class:`KaolinDatasetItem` with the fields:
 
-            * ``data``: containing a namedtuple returned by :func:`kaolin.io.off.import_mesh`.
+            * ``data``: containing a :class:`kaolin.rep.SurfaceMesh` returned by :func:`kaolin.io.obj.import_mesh`.
             * ``attributes``: containing a dictionary with the following key-value pairs:
 
                 * 'name': the model name (i.e the subfolder name)