Fix test

MatsMoll · May 22, 2024 · dbfd30d · dbfd30d
1 parent cbf2f9d
commit dbfd30d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -26,10 +26,10 @@ jobs:
           token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Build docker images
-        run: docker-compose -f docker-compose.yaml build base-image && docker-compose -f docker-compose.yaml build test-action
+        run: docker-compose -f docker-compose.yaml build base-image && docker-compose -f docker-compose.yaml build test
 
       - name: Test
-        run: docker-compose -f docker-compose.yaml run test-action
+        run: docker-compose -f docker-compose.yaml run test
 
   generate-main-contract:
     runs-on: ubuntu-22.04

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -96,7 +96,7 @@ services:
     depends_on:
       - base-image
 
-  test-action:
+  test:
     image: project-dev
     build:
       context: .
@@ -109,10 +109,10 @@ services:
     depends_on:
       - base-image
 
+  #   # Uncomment below to expose Ollama API outside the container stack
   # ollama:
   #   container_name: ollama
   #   image: ollama/ollama:latest
-  #   # Uncomment below to expose Ollama API outside the container stack
   #   pull_policy: always
   #   tty: true
   #   restart: unless-stopped

diff --git a/src/pipelines/train.py b/src/pipelines/train.py
@@ -42,7 +42,7 @@ async def generate_train_test(
             ) # type: ignore
         )
     )
-    if model_store.dataset_store:
+    if model_store.dataset_store and dataset_dir:
 
         if dataset_id is None:
             dataset_id = flow_run.get_name()