From a3829b0fba30b12481a7b00e5c08f08901173921 Mon Sep 17 00:00:00 2001 From: skoulouzis Date: Mon, 10 Feb 2025 14:19:11 +0100 Subject: [PATCH] fix get --- app/services/cell_extractor/header_extractor.py | 6 +++--- app/services/cell_extractor/r_header_extractor.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/app/services/cell_extractor/header_extractor.py b/app/services/cell_extractor/header_extractor.py index 2aa86ed..b6251b0 100644 --- 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 index c17db5a..c6f0adb 100644 --- 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 = []