added dataset to study fields

BioImage-Archive · Jun 21, 2024 · f48c368 · f48c368
1 parent 8c39ea6
commit f48c368
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/bia-shared-datamodels/src/bia_models/bia_data_model.py b/bia-shared-datamodels/src/bia_models/bia_data_model.py
@@ -38,6 +38,7 @@ class ExperimentalImagingDataset(
 ):
     image: List[UUID] = Field()
     file: List[UUID] = Field()
+    submitted_in_study: UUID = Field()
     # we include image analysis and correlation
 
 
@@ -84,6 +85,7 @@ class ImageAnnotationDataset(
 ):
     image: List[UUID] = Field()
     file: List[UUID] = Field()
+    submitted_in_study: UUID = Field()
 
 
 class AnnotationFileReference(
@@ -92,3 +94,4 @@ class AnnotationFileReference(
 ):
     represenatation: List[UUID] = Field()
     source_image: UUID = Field()
+    submission_dataset: UUID = Field()
diff --git a/bia-shared-datamodels/src/bia_models/semantic_models.py b/bia-shared-datamodels/src/bia_models/semantic_models.py
@@ -206,6 +206,9 @@ class DatasetMixin(BaseModel):
     file_reference_count: int = Field(
         description="""Number of files associated with the study."""
     )
+    submitted_in_study: Study = Field(
+        description="""The study the dataset was submitted in."""
+    )
 
 
 class FileReference(BaseModel):