Minor fixes

2022-07-02 21:57:09 +02:00 · 2022-07-02 21:57:09 +02:00 · 2ba4690e12
commit 2ba4690e12
parent 31f5bab2f4
1 changed files with 9 additions and 9 deletions
--- a/src/great_ai/great_ai/tracing/add_ground_truth.py
+++ b/src/great_ai/great_ai/tracing/add_ground_truth.py
@ -20,8 +20,8 @@ def add_ground_truth(
    expected_outputs: Iterable[T],
    *,
    tags: List[str] = [],
-    train_split_ratio: float,
-    test_split_ratio: float,
+    train_split_ratio: float = 1,
+    test_split_ratio: float = 0,
    validation_split_ratio: float = 0
 ) -> None:
    get_context()  # this resets the seed