add dataset

Clinical-Genomics · May 2, 2022 · 8f81332 · 8f81332
1 parent 4c74009
commit 8f81332
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/statina/API/v2/endpoints/sample.py b/statina/API/v2/endpoints/sample.py
@@ -40,10 +40,11 @@ def samples(
         **sample_query.dict(), adapter=adapter
     )
     validated_samples: List[SampleValidator] = []
-    for sample in database_samples:
+    for database_sample in database_samples:
         validated_samples.append(
             SampleValidator(
-                **sample.dict(), dataset=get_dataset(adapter=adapter, batch_id=sample.batch_id)
+                **database_sample.dict(),
+                dataset=get_dataset(adapter=adapter, batch_id=database_sample.batch_id),
             )
         )
     samples: List[SampleResponse] = [

diff --git a/statina/crud/find/datasets.py b/statina/crud/find/datasets.py
@@ -14,7 +14,7 @@ def get_dataset(
     if batch_id:
         batch: dict = adapter.batch_collection.find_one({"batch_id": batch_id})
         if not batch:
-            return base_dataset_thresholds
+            return None
         if not batch.get("dataset"):
             return base_dataset_thresholds
         dataset = adapter.dataset_collection.find({"name": batch.get("dataset")})