RandomRotationVectorDataset

Genentech · May 22, 2024 · 036913d · 036913d
1 parent dd7ef6e
commit 036913d
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/src/beignet/datasets/_random_euler_angle_dataset.py b/src/beignet/datasets/_random_euler_angle_dataset.py
@@ -23,6 +23,10 @@ def __init__(
         requires_grad: bool | None = False,
         transform: Callable | Transform | None = None,
     ):
+        r"""
+        Parameters
+        ----------
+        """
         super().__init__(
             beignet.random_euler_angle(
                 size,

diff --git a/src/beignet/datasets/_random_quaternion_dataset.py b/src/beignet/datasets/_random_quaternion_dataset.py
@@ -22,6 +22,10 @@ def __init__(
         requires_grad: bool | None = False,
         transform: Callable | Transform | None = None,
     ):
+        r"""
+        Parameters
+        ----------
+        """
         super().__init__(
             beignet.random_quaternion(
                 size,

diff --git a/src/beignet/datasets/_random_rotation_matrix_dataset.py b/src/beignet/datasets/_random_rotation_matrix_dataset.py
@@ -20,6 +20,10 @@ def __init__(
         requires_grad: bool | None = False,
         transform: Callable | Transform | None = None,
     ):
+        r"""
+        Parameters
+        ----------
+        """
         super().__init__(
             beignet.random_rotation_matrix(
                 size,

diff --git a/src/beignet/datasets/_random_rotation_vector_dataset.py b/src/beignet/datasets/_random_rotation_vector_dataset.py
@@ -21,6 +21,10 @@ def __init__(
         requires_grad: bool | None = False,
         transform: Callable | Transform | None = None,
     ):
+        r"""
+        Parameters
+        ----------
+        """
         super().__init__(
             beignet.random_rotation_vector(
                 size,