fix get

NaaVRE · Feb 10, 2025 · a3829b0 · a3829b0
1 parent 5325935
commit a3829b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app/services/cell_extractor/header_extractor.py b/app/services/cell_extractor/header_extractor.py
@@ -224,7 +224,7 @@ def _infer_cell_interface_vars(
         if header is None:
             return []
         vars = []
-        items = header['NaaVRE']['cell'].get_base_image_tags(item_type)
+        items = header['NaaVRE']['cell'].get(item_type)
         if items is None:
             return []
         for item in items:
@@ -263,7 +263,7 @@ def get_cell_secrets(self) -> list[dict]:
     def get_cell_confs(self) -> list[dict]:
         if self.cell_header is None:
             return []
-        items = self.cell_header['NaaVRE']['cell'].get_base_image_tags('confs')
+        items = self.cell_header['NaaVRE']['cell'].get('confs')
         if items is None:
             return []
         confs = []
@@ -283,7 +283,7 @@ def get_cell_confs(self) -> list[dict]:
     def get_cell_dependencies(self, confs) -> list[dict]:
         if self.cell_header is None:
             return []
-        items = self.cell_header['NaaVRE']['cell'].get_base_image_tags(
+        items = self.cell_header['NaaVRE']['cell'].get(
             'dependencies')
         if items is None:
             return []

diff --git a/app/services/cell_extractor/r_header_extractor.py b/app/services/cell_extractor/r_header_extractor.py
@@ -9,7 +9,7 @@ def __init__(self, notebook_data, base_image_tags_url: str):
     def get_cell_confs(self) -> list[dict]:
         if self.cell_header is None:
             return []
-        items = self.cell_header['NaaVRE']['cell'].get_base_image_tags('confs')
+        items = self.cell_header['NaaVRE']['cell'].get('confs')
         if items is None:
             return []
         confs = []