From ce8f6a1418a94681b84d06ce8f3fbbe4425864ef Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= Date: Mon, 1 May 2023 10:55:16 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20test:=20level=202?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/options/generate/generator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py index f9a6539..a9f2bc9 100644 --- a/src/options/generate/generator.py +++ b/src/options/generate/generator.py @@ -219,7 +219,9 @@ metas: return {DOCKER_FILE_NAME: docker_file_template.replace('{{apt_get_packages}}', '{apt_get_packages}').format(apt_get_packages=content_raw)} def parse_result_fn_requirements(self, content_raw: str): - lines = content_raw.split('\n') + content_parsed = self.extract_content_from_result(content_raw, 'requirements.txt', match_single_block=True) + + lines = content_parsed.split('\n') lines = [line for line in lines if not any([pkg in line for pkg in ['jina', 'docarray', 'openai', 'pytest']])] content_modified = f'''jina==3.15.1.dev14 docarray==0.21.0