🐛 Fix SampleRelationship target ID lookup

kids-first · Jun 12, 2024 · f1fc3e0 · f1fc3e0
1 parent ed6c8aa
commit f1fc3e0
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/kf_lib_data_ingest/target_api_plugins/kids_first_dataservice.py b/kf_lib_data_ingest/target_api_plugins/kids_first_dataservice.py
@@ -520,8 +520,12 @@ def get_key_components(cls, record, get_target_id_from_record):
                     Sample,
                     {
                         CONCEPT.SAMPLE.ID: record[
-                            CONCEPT.SAMPLE_RELATIONSHIP.PARENT_SAMPLE
-                        ]
+                            CONCEPT.SAMPLE_RELATIONSHIP.PARENT_SAMPLE.ID
+                        ],
+                        CONCEPT.SAMPLE.TARGET_SERVICE_ID: record[
+                            CONCEPT.SAMPLE_RELATIONSHIP.PARENT_SAMPLE.TARGET_SERVICE_ID
+                        ],
+
                     },
                 )
             ),
@@ -530,8 +534,12 @@ def get_key_components(cls, record, get_target_id_from_record):
                     Sample,
                     {
                         CONCEPT.SAMPLE.ID: record[
-                            CONCEPT.SAMPLE_RELATIONSHIP.CHILD_SAMPLE
-                        ]
+                            CONCEPT.SAMPLE_RELATIONSHIP.CHILD_SAMPLE.ID
+                        ],
+                        CONCEPT.SAMPLE.TARGET_SERVICE_ID: record[
+                            CONCEPT.SAMPLE_RELATIONSHIP.CHILD_SAMPLE.TARGET_SERVICE_ID
+                        ],
+
                     },
                 )
             ),
@@ -1111,7 +1119,8 @@ class SequencingExperimentGenomicFile:
     class_name = "sequencing_experiment_genomic_file"
     api_path = "sequencing-experiment-genomic-files"
     target_id_concept = CONCEPT.SEQUENCING_GENOMIC_FILE.TARGET_SERVICE_ID
-    service_id_fields = {"kf_id", "sequencing_experiment_id", "genomic_file_id"}
+    service_id_fields = {
+        "kf_id", "sequencing_experiment_id", "genomic_file_id"}
 
     @classmethod
     def get_key_components(cls, record, get_target_id_from_record):