From 704ed6fedc6bc221f459a0ea4796c91097dce105 Mon Sep 17 00:00:00 2001 From: SdgJlbl Date: Thu, 3 Oct 2024 10:42:59 +0200 Subject: [PATCH] fix: binary dependencies test Signed-off-by: SdgJlbl --- substrafl/remote/register/register.py | 3 +-- tests/dependency/test_dependency.py | 3 ++- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/substrafl/remote/register/register.py b/substrafl/remote/register/register.py index d9bbb0f0..55745fb8 100644 --- a/substrafl/remote/register/register.py +++ b/substrafl/remote/register/register.py @@ -43,8 +43,7 @@ RUN apt-get update -y\ && pip uninstall -y setuptools\ && apt-get install -y {binary_dependencies}\ - && apt-get clean\ - && rm -rf /var/lib/apt/lists/* + && apt-get clean """ _GPU_BASE_IMAGE = """ diff --git a/tests/dependency/test_dependency.py b/tests/dependency/test_dependency.py index a8b30f87..660d4591 100644 --- a/tests/dependency/test_dependency.py +++ b/tests/dependency/test_dependency.py @@ -307,7 +307,8 @@ def test_binary_dependencies( client = network.clients[0] algo_deps = Dependency( - binary_dependencies=["gcc"], + pypi_dependencies=["pytest"], + binary_dependencies=["python3-dev"], editable_mode=True, ) function_key = self._register_function(dummy_algo_class(), algo_deps, client, session_dir)