Skip to content

Commit

Permalink
Fix copy base to folder
Browse files Browse the repository at this point in the history
Using absolute path instead of relative
  • Loading branch information
satyaog committed Sep 5, 2024
1 parent 0016cd7 commit 1e62419
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions milabench/multi.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
from .config import get_base_folder
from .pack import Package
from .remote import (
INSTALL_FOLDER,
is_main_local,
is_multinode,
is_remote,
Expand Down Expand Up @@ -92,16 +91,15 @@ async def copy_base_to_workers(setup):
print("Coping main setup from this node to worker")
# copy the main setup to the workers
# so it copies the bench venv already, no need for python
from milabench.remote import copy_folder
from milabench.system import SystemConfig
from milabench.remote import INSTALL_FOLDER, copy_folder

# we copy the entire content of base
# FIXME: handle custom (venv, cache, data, etc...) directories
#
copy_plan = ListCommand(
*[
copy_folder(setup, _dir)
for _dir in [INSTALL_FOLDER, SystemConfig().base]
for _dir in [INSTALL_FOLDER, str(setup.dirs.base)]
]
)
remote_task = asyncio.create_task(copy_plan.execute())
Expand Down

0 comments on commit 1e62419

Please sign in to comment.