minor change

jitingxu1 · Jul 17, 2024 · 85a3921 · 85a3921
1 parent 2fab5f5
commit 85a3921
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/docs/_quarto.yml b/docs/_quarto.yml
@@ -86,7 +86,7 @@ website:
 
         - section: Utilities
           contents:
-            - reference/train-test-split.qmd
+            - reference/utils-train-test-split.qmd
 
 format:
   html:
@@ -234,9 +234,9 @@ quartodoc:
       package: ibis_ml
       contents:
         - kind: page
-          path: train-test-split
+          path: utils-train-test-split
           summary:
-            name: Train test split
+            name: Train-test split
             desc: Randomly split Ibis table
           contents:
             - train_test_split
diff --git a/ibis_ml/utils/_train_test_split.py b/ibis_ml/utils/_train_test_split.py
@@ -46,11 +46,11 @@ def train_test_split(
 
     Returns
     -------
-    tuple(ir.Table, ir.Table)
+    tuple[ir.Table, ir.Table]
         A tuple containing two Ibis tables: (train_table, test_table).
 
     Raises
-    ----------
+    ------
     ValueError
         If test_size is not a float between 0 and 1.
 
@@ -74,6 +74,7 @@ def train_test_split(
     # Set the random seed for reproducibility
     if random_seed:
         random.seed(random_seed)
+
     # Generate a random 256-bit key
     random_key = str(random.getrandbits(256))