docs

sdv-dev · Jan 23, 2024 · ab8f912 · ab8f912
1 parent b81fc6e
commit ab8f912
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ctgan/data_sampler.py b/ctgan/data_sampler.py
@@ -128,7 +128,10 @@ def sample_data(self, data, n, col, opt):
         """Sample data from original training data satisfying the sampled conditional vector.
 
         Returns:
-            n rows of matrix data.
+            data:
+                The training data.
+            n:
+                n rows of matrix data.
         """
         if col is None:
             idx = np.random.randint(len(data), size=n)