From a11eed023c70aff3813ae050046e041cddf31375 Mon Sep 17 00:00:00 2001
From: Harshad Hegde <hegdehb@gmail.com>
Date: Mon, 26 Aug 2024 11:47:50 -0500
Subject: [PATCH] Added NCBITaxon:XX edge to strain

---
 kg_microbe/transform_utils/mediadive/mediadive.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/kg_microbe/transform_utils/mediadive/mediadive.py b/kg_microbe/transform_utils/mediadive/mediadive.py
index df8a2eff..3a840d32 100644
--- a/kg_microbe/transform_utils/mediadive/mediadive.py
+++ b/kg_microbe/transform_utils/mediadive/mediadive.py
@@ -84,6 +84,7 @@
     NCBI_CATEGORY,
     NCBI_TO_MEDIUM_EDGE,
     NCBITAXON_ID_COLUMN,
+    NCBITAXON_PREFIX,
     OBJECT_ID_COLUMN,
     PUBCHEM_KEY,
     PUBCHEM_PREFIX,
@@ -387,9 +388,21 @@ def run(self, data_file: Union[Optional[Path], Optional[str]] = None, show_statu
                                                 medium_id,
                                                 IS_GROWN_IN,
                                                 strain_id,
-                                            ]
+                                            ],
                                         ]
                                     )
+                                    if strain.get(ID_COLUMN):
+                                        medium_strain_edge.extend(
+                                            [
+                                                [
+                                                    ncbi_strain_id,
+                                                    SUBCLASS_PREDICATE,
+                                                    NCBITAXON_PREFIX+str(strain.get(ID_COLUMN)),
+                                                    RDFS_SUBCLASS_OF,
+                                                    strain_id
+                                                ],
+                                            ]
+                                        )
                                     edge_writer.writerows(medium_strain_edge)
 
                     if SOLUTIONS_KEY not in json_obj: