Minor fixes

.github/workflows/publish.yaml

@@ -11,10 +11,10 @@ jobs:
     steps:
       - uses: actions/checkout@v3
 
-      - name: Set up Python 3.9.2
+      - name: Set up Python 3.9
         uses: actions/setup-python@v2
         with:
-          python-version: 3.9.2
+          python-version: 3.9
 
       - name: Install pypa/build
         run: python -m pip install build --user

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = great-ai
-version = 0.0.8
+version = 0.0.9
 author = András Schmelczer
 author_email = andras@scoutinscience.com
 description = 

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

src/great_ai/utilities/parallel_map/parallel_map.py

@@ -88,8 +88,8 @@ def parallel_map(
     ctx = mp.get_context("spawn")
     ctx.freeze_support()
 
-    input_queue = ctx.Queue(0 if config.chunk_count is None else config.chunk_count)
-    output_queue = ctx.Queue(0 if config.chunk_count is None else config.chunk_count)
+    input_queue = ctx.Queue(config.concurrency * 2)
+    output_queue = ctx.Queue(config.concurrency * 2)
     should_stop = ctx.Event()
 
     processes = [

src/great_ai/utilities/parallel_map/threaded_parallel_map.py

@@ -86,8 +86,8 @@ def threaded_parallel_map(
             ignore_exceptions=ignore_exceptions,
         )
 
-    input_queue = queue.Queue(0 if config.chunk_count is None else config.chunk_count)
-    output_queue = queue.Queue(0 if config.chunk_count is None else config.chunk_count)
+    input_queue = queue.Queue(config.concurrency * 2)
+    output_queue = queue.Queue(config.concurrency * 2)
     should_stop = threading.Event()
 
     threads = [