fractal-server 2.9.0a2__tar.gz → 2.9.0a3__tar.gz

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fractal-server
-Version: 2.9.0a2
+Version: 2.9.0a3
 Summary: Server component of the Fractal analytics platform
 Home-page: https://github.com/fractal-analytics-platform/fractal-server
 License: BSD-3-Clause

fractal_server-2.9.0a3/fractal_server/__init__.py

@@ -0,0 +1 @@
+__VERSION__ = "2.9.0a3"

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/routes/api/v2/task_collection.py

@@ -37,6 +37,7 @@ from fractal_server.app.schemas.v2 import TaskGroupV2OriginEnum
 from fractal_server.tasks.v2.local.collect import (
     collect_local,
 )
+from fractal_server.tasks.v2.ssh import collect_ssh
 from fractal_server.tasks.v2.utils_package_names import _parse_wheel_filename
 from fractal_server.tasks.v2.utils_package_names import normalize_package_name
 from fractal_server.tasks.v2.utils_python_interpreter import (
@@ -63,10 +64,7 @@ async def collect_tasks_pip(
     db: AsyncSession = Depends(get_async_db),
 ) -> TaskGroupActivityV2Read:
     """
-    Task collection endpoint
-
-    Trigger the creation of a dedicated virtual environment, the installation
-    of a package and the collection of tasks as advertised in the manifest.
+    Task-collection endpoint
     """
 
     # Get settings
@@ -246,12 +244,7 @@ async def collect_tasks_pip(
 
     if settings.FRACTAL_RUNNER_BACKEND == "slurm_ssh":
         # SSH task collection
-
-        from fractal_server.tasks.v2.ssh.collect import (
-            collect_ssh,
-        )
-
-        # User appropriate FractalSSH object
+        # Use appropriate FractalSSH object
         ssh_credentials = dict(
             user=user_settings.ssh_username,
             host=user_settings.ssh_host,

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/routes/api/v2/task_group_lifecycle.py

@@ -2,9 +2,11 @@ from fastapi import APIRouter
 from fastapi import BackgroundTasks
 from fastapi import Depends
 from fastapi import HTTPException
+from fastapi import Request
 from fastapi import Response
 from fastapi import status
 
+from ...aux.validate_user_settings import validate_user_settings
 from ._aux_functions_task_lifecycle import check_no_ongoing_activity
 from ._aux_functions_tasks import _get_task_group_full_access
 from fractal_server.app.db import AsyncSession
@@ -22,6 +24,8 @@ from fractal_server.logger import set_logger
 from fractal_server.syringe import Inject
 from fractal_server.tasks.v2.local import deactivate_local
 from fractal_server.tasks.v2.local import reactivate_local
+from fractal_server.tasks.v2.ssh import deactivate_ssh
+from fractal_server.tasks.v2.ssh import reactivate_ssh
 from fractal_server.utils import get_timestamp
 
 router = APIRouter()
@@ -38,6 +42,7 @@ async def deactivate_task_group(
     task_group_id: int,
     background_tasks: BackgroundTasks,
     response: Response,
+    request: Request,
     user: UserOAuth = Depends(current_active_user),
     db: AsyncSession = Depends(get_async_db),
 ) -> TaskGroupReadV2:
@@ -103,10 +108,29 @@ async def deactivate_task_group(
     # Submit background task
     settings = Inject(get_settings)
     if settings.FRACTAL_RUNNER_BACKEND == "slurm_ssh":
-        raise HTTPException(
-            status_code=status.HTTP_501_NOT_IMPLEMENTED,
-            detail="Not implemented (yet) for SSH.",
+
+        # Validate user settings (backend-specific)
+        user_settings = await validate_user_settings(
+            user=user, backend=settings.FRACTAL_RUNNER_BACKEND, db=db
+        )
+
+        # User appropriate FractalSSH object
+        ssh_credentials = dict(
+            user=user_settings.ssh_username,
+            host=user_settings.ssh_host,
+            key_path=user_settings.ssh_private_key_path,
+        )
+        fractal_ssh_list = request.app.state.fractal_ssh_list
+        fractal_ssh = fractal_ssh_list.get(**ssh_credentials)
+
+        background_tasks.add_task(
+            deactivate_ssh,
+            task_group_id=task_group.id,
+            task_group_activity_id=task_group_activity.id,
+            fractal_ssh=fractal_ssh,
+            tasks_base_dir=user_settings.ssh_tasks_dir,
         )
+
     else:
         background_tasks.add_task(
             deactivate_local,
@@ -130,6 +154,7 @@ async def reactivate_task_group(
     task_group_id: int,
     background_tasks: BackgroundTasks,
     response: Response,
+    request: Request,
     user: UserOAuth = Depends(current_active_user),
     db: AsyncSession = Depends(get_async_db),
 ) -> TaskGroupReadV2:
@@ -203,10 +228,29 @@ async def reactivate_task_group(
     # Submit background task
     settings = Inject(get_settings)
     if settings.FRACTAL_RUNNER_BACKEND == "slurm_ssh":
-        raise HTTPException(
-            status_code=status.HTTP_501_NOT_IMPLEMENTED,
-            detail="Not implemented (yet) for SSH.",
+
+        # Validate user settings (backend-specific)
+        user_settings = await validate_user_settings(
+            user=user, backend=settings.FRACTAL_RUNNER_BACKEND, db=db
+        )
+
+        # Use appropriate FractalSSH object
+        ssh_credentials = dict(
+            user=user_settings.ssh_username,
+            host=user_settings.ssh_host,
+            key_path=user_settings.ssh_private_key_path,
+        )
+        fractal_ssh_list = request.app.state.fractal_ssh_list
+        fractal_ssh = fractal_ssh_list.get(**ssh_credentials)
+
+        background_tasks.add_task(
+            reactivate_ssh,
+            task_group_id=task_group.id,
+            task_group_activity_id=task_group_activity.id,
+            fractal_ssh=fractal_ssh,
+            tasks_base_dir=user_settings.ssh_tasks_dir,
         )
+
     else:
         background_tasks.add_task(
             reactivate_local,

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/_validators.py

@@ -1,6 +1,4 @@
 import os
-from datetime import datetime
-from datetime import timezone
 from typing import Any
 from typing import Optional
 
@@ -103,15 +101,3 @@ def val_unique_list(attribute: str):
         return must_be_unique
 
     return val
-
-
-def valutc(attribute: str):
-    def val(timestamp: Optional[datetime]) -> Optional[datetime]:
-        """
-        Replace `tzinfo` with `timezone.utc`.
-        """
-        if timestamp is not None:
-            return timestamp.replace(tzinfo=timezone.utc)
-        return None
-
-    return val

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v1/applyworkflow.py

@@ -7,7 +7,6 @@ from pydantic import validator
 from pydantic.types import StrictStr
 
 from .._validators import valstr
-from .._validators import valutc
 from .dumps import DatasetDumpV1
 from .dumps import ProjectDumpV1
 from .dumps import WorkflowDumpV1
@@ -150,13 +149,6 @@ class ApplyWorkflowReadV1(_ApplyWorkflowBaseV1):
     first_task_index: Optional[int]
     last_task_index: Optional[int]
 
-    _start_timestamp = validator("start_timestamp", allow_reuse=True)(
-        valutc("start_timestamp")
-    )
-    _end_timestamp = validator("end_timestamp", allow_reuse=True)(
-        valutc("end_timestamp")
-    )
-
 
 class ApplyWorkflowUpdateV1(BaseModel):
     """

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v1/dataset.py

@@ -8,7 +8,6 @@ from pydantic import validator
 
 from .._validators import val_absolute_path
 from .._validators import valstr
-from .._validators import valutc
 from .dumps import WorkflowTaskDumpV1
 from .project import ProjectReadV1
 from .workflow import WorkflowTaskStatusTypeV1
@@ -151,10 +150,6 @@ class DatasetReadV1(_DatasetBaseV1):
     project: ProjectReadV1
     timestamp_created: datetime
 
-    _timestamp_created = validator("timestamp_created", allow_reuse=True)(
-        valutc("timestamp_created")
-    )
-
 
 class DatasetStatusReadV1(BaseModel):
     """

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v1/project.py

@@ -5,7 +5,6 @@ from pydantic import BaseModel
 from pydantic import validator
 
 from .._validators import valstr
-from .._validators import valutc
 
 
 __all__ = (
@@ -50,10 +49,6 @@ class ProjectReadV1(_ProjectBaseV1):
     id: int
     timestamp_created: datetime
 
-    _timestamp_created = validator("timestamp_created", allow_reuse=True)(
-        valutc("timestamp_created")
-    )
-
 
 class ProjectUpdateV1(_ProjectBaseV1):
     """

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v1/state.py

@@ -3,9 +3,6 @@ from typing import Any
 from typing import Optional
 
 from pydantic import BaseModel
-from pydantic import validator
-
-from fractal_server.app.schemas._validators import valutc
 
 
 class StateRead(BaseModel):
@@ -19,5 +16,3 @@ class StateRead(BaseModel):
     id: Optional[int]
     data: dict[str, Any]
     timestamp: datetime
-
-    _timestamp = validator("timestamp", allow_reuse=True)(valutc("timestamp"))

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v1/workflow.py

@@ -8,7 +8,6 @@ from pydantic import validator
 
 from .._validators import valint
 from .._validators import valstr
-from .._validators import valutc
 from .project import ProjectReadV1
 from .task import TaskExportV1
 from .task import TaskImportV1
@@ -135,10 +134,6 @@ class WorkflowReadV1(_WorkflowBaseV1):
     project: ProjectReadV1
     timestamp_created: datetime
 
-    _timestamp_created = validator("timestamp_created", allow_reuse=True)(
-        valutc("timestamp_created")
-    )
-
 
 class WorkflowCreateV1(_WorkflowBaseV1):
     """

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v2/dataset.py

@@ -7,7 +7,6 @@ from pydantic import Field
 from pydantic import validator
 
 from .._validators import valstr
-from .._validators import valutc
 from .dumps import WorkflowTaskDumpV2
 from .project import ProjectReadV2
 from .workflowtask import WorkflowTaskStatusTypeV2
@@ -62,11 +61,6 @@ class DatasetReadV2(BaseModel):
     zarr_dir: str
     filters: Filters = Field(default_factory=Filters)
 
-    # Validators
-    _timestamp_created = validator("timestamp_created", allow_reuse=True)(
-        valutc("timestamp_created")
-    )
-
 
 class DatasetUpdateV2(BaseModel, extra=Extra.forbid):
 

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v2/job.py

@@ -8,7 +8,6 @@ from pydantic import validator
 from pydantic.types import StrictStr
 
 from .._validators import valstr
-from .._validators import valutc
 from .dumps import DatasetDumpV2
 from .dumps import ProjectDumpV2
 from .dumps import WorkflowDumpV2
@@ -101,13 +100,6 @@ class JobReadV2(BaseModel):
     last_task_index: Optional[int]
     worker_init: Optional[str]
 
-    _start_timestamp = validator("start_timestamp", allow_reuse=True)(
-        valutc("start_timestamp")
-    )
-    _end_timestamp = validator("end_timestamp", allow_reuse=True)(
-        valutc("end_timestamp")
-    )
-
 
 class JobUpdateV2(BaseModel, extra=Extra.forbid):
 

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v2/project.py

@@ -6,7 +6,6 @@ from pydantic import Extra
 from pydantic import validator
 
 from .._validators import valstr
-from .._validators import valutc
 
 
 class ProjectCreateV2(BaseModel, extra=Extra.forbid):
@@ -21,10 +20,6 @@ class ProjectReadV2(BaseModel):
     id: int
     name: str
     timestamp_created: datetime
-    # Validators
-    _timestamp_created = validator("timestamp_created", allow_reuse=True)(
-        valutc("timestamp_created")
-    )
 
 
 class ProjectUpdateV2(BaseModel, extra=Extra.forbid):

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/app/schemas/v2/workflow.py

@@ -6,7 +6,6 @@ from pydantic import Extra
 from pydantic import validator
 
 from .._validators import valstr
-from .._validators import valutc
 from .project import ProjectReadV2
 from .workflowtask import WorkflowTaskExportV2
 from .workflowtask import WorkflowTaskImportV2
@@ -31,10 +30,6 @@ class WorkflowReadV2(BaseModel):
     project: ProjectReadV2
     timestamp_created: datetime
 
-    _timestamp_created = validator("timestamp_created", allow_reuse=True)(
-        valutc("timestamp_created")
-    )
-
 
 class WorkflowReadV2WithWarnings(WorkflowReadV2):
     task_list: list[WorkflowTaskReadV2WithWarning]

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/ssh/_fabric.py

@@ -321,8 +321,10 @@ class FractalSSH(object):
                     f"{prefix} END   running '{cmd}' over SSH, "
                     f"elapsed {t_1-t_0:.3f}"
                 )
-                self.logger.debug(f"STDOUT: {res.stdout}")
-                self.logger.debug(f"STDERR: {res.stderr}")
+                self.logger.debug("STDOUT:")
+                self.logger.debug(res.stdout)
+                self.logger.debug("STDERR:")
+                self.logger.debug(res.stderr)
                 return res.stdout
             except NoValidConnectionsError as e:
                 # Case 2: Command fails with a connection error

fractal_server-2.9.0a2/fractal_server/tasks/v2/local/utils_local.py → fractal_server-2.9.0a3/fractal_server/tasks/v2/local/_utils.py

@@ -1,5 +1,6 @@
 from pathlib import Path
 
+from fractal_server.app.schemas.v2 import TaskCreateV2
 from fractal_server.logger import get_logger
 from fractal_server.tasks.v2.utils_templates import customize_template
 from fractal_server.utils import execute_command_sync
@@ -43,3 +44,27 @@ def _customize_and_run_template(
     stdout = execute_command_sync(command=cmd, logger_name=logger_name)
     logger.debug(f"_customize_and_run_template {template_filename} - END")
     return stdout
+
+
+def check_task_files_exist(task_list: list[TaskCreateV2]) -> None:
+    """
+    Check that the modules listed in task commands point to existing files.
+
+    Args:
+        task_list:
+    """
+    for _task in task_list:
+        if _task.command_non_parallel is not None:
+            _task_path = _task.command_non_parallel.split()[1]
+            if not Path(_task_path).exists():
+                raise FileNotFoundError(
+                    f"Task `{_task.name}` has `command_non_parallel` "
+                    f"pointing to missing file `{_task_path}`."
+                )
+        if _task.command_parallel is not None:
+            _task_path = _task.command_parallel.split()[1]
+            if not Path(_task_path).exists():
+                raise FileNotFoundError(
+                    f"Task `{_task.name}` has `command_parallel` "
+                    f"pointing to missing file `{_task_path}`."
+                )

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/tasks/v2/local/collect.py

@@ -6,7 +6,7 @@ from pathlib import Path
 from tempfile import TemporaryDirectory
 
 from ..utils_database import create_db_tasks_and_update_task_group
-from .utils_local import _customize_and_run_template
+from ._utils import _customize_and_run_template
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models.v2 import TaskGroupActivityV2
 from fractal_server.app.models.v2 import TaskGroupV2
@@ -16,9 +16,9 @@ from fractal_server.app.schemas.v2.manifest import ManifestV2
 from fractal_server.logger import get_logger
 from fractal_server.logger import set_logger
 from fractal_server.tasks.utils import get_log_path
+from fractal_server.tasks.v2.local._utils import check_task_files_exist
 from fractal_server.tasks.v2.utils_background import _prepare_tasks_metadata
 from fractal_server.tasks.v2.utils_background import add_commit_refresh
-from fractal_server.tasks.v2.utils_background import check_task_files_exist
 from fractal_server.tasks.v2.utils_background import fail_and_cleanup
 from fractal_server.tasks.v2.utils_background import get_current_log
 from fractal_server.tasks.v2.utils_package_names import compare_package_names

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/tasks/v2/local/deactivate.py

@@ -7,7 +7,7 @@ from tempfile import TemporaryDirectory
 from ..utils_background import add_commit_refresh
 from ..utils_background import fail_and_cleanup
 from ..utils_templates import get_collection_replacements
-from .utils_local import _customize_and_run_template
+from ._utils import _customize_and_run_template
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models.v2 import TaskGroupActivityV2
 from fractal_server.app.models.v2 import TaskGroupV2

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/tasks/v2/local/reactivate.py

@@ -7,7 +7,7 @@ from tempfile import TemporaryDirectory
 from ..utils_background import add_commit_refresh
 from ..utils_background import fail_and_cleanup
 from ..utils_templates import get_collection_replacements
-from .utils_local import _customize_and_run_template
+from ._utils import _customize_and_run_template
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models.v2 import TaskGroupActivityV2
 from fractal_server.app.models.v2 import TaskGroupV2

fractal_server-2.9.0a3/fractal_server/tasks/v2/ssh/__init__.py

@@ -0,0 +1,3 @@
+from .collect import collect_ssh  # noqa
+from .deactivate import deactivate_ssh  # noqa
+from .reactivate import reactivate_ssh  # noqa

fractal_server-2.9.0a3/fractal_server/tasks/v2/ssh/_utils.py

@@ -0,0 +1,87 @@
+import os
+from pathlib import Path
+
+from fractal_server.app.models.v2 import TaskGroupV2
+from fractal_server.logger import get_logger
+from fractal_server.ssh._fabric import FractalSSH
+from fractal_server.tasks.v2.utils_templates import customize_template
+
+
+def _customize_and_run_template(
+    *,
+    template_filename: str,
+    replacements: list[tuple[str, str]],
+    script_dir_local: str,
+    prefix: str,
+    fractal_ssh: FractalSSH,
+    script_dir_remote: str,
+    logger_name: str,
+) -> str:
+    """
+    Customize one of the template bash scripts, transfer it to the remote host
+    via SFTP and then run it via SSH.
+
+    Args:
+        template_filename: Filename of the template file (ends with ".sh").
+        replacements: Dictionary of replacements.
+        script_dir: Local folder where the script will be placed.
+        prefix: Prefix for the script filename.
+        fractal_ssh: FractalSSH object
+        script_dir_remote: Remote scripts directory
+    """
+    logger = get_logger(logger_name=logger_name)
+    logger.debug(f"_customize_and_run_template {template_filename} - START")
+    # Prepare name and path of script
+    if not template_filename.endswith(".sh"):
+        raise ValueError(
+            f"Invalid {template_filename=} (it must end with '.sh')."
+        )
+    script_filename = f"{prefix}_{template_filename}"
+    script_path_local = (Path(script_dir_local) / script_filename).as_posix()
+
+    customize_template(
+        template_name=template_filename,
+        replacements=replacements,
+        script_path=script_path_local,
+    )
+
+    # Transfer script to remote host
+    script_path_remote = os.path.join(
+        script_dir_remote,
+        script_filename,
+    )
+    logger.debug(f"Now transfer {script_path_local=} over SSH.")
+    fractal_ssh.send_file(
+        local=script_path_local,
+        remote=script_path_remote,
+    )
+
+    # Execute script remotely
+    cmd = f"bash {script_path_remote}"
+    logger.debug(f"Now run '{cmd}' over SSH.")
+    stdout = fractal_ssh.run_command(cmd=cmd)
+
+    logger.debug(f"_customize_and_run_template {template_filename} - END")
+    return stdout
+
+
+def _copy_wheel_file_ssh(
+    *, task_group: TaskGroupV2, fractal_ssh: FractalSSH, logger_name: str
+) -> str:
+    """
+    Handle the situation where `task_group.wheel_path` is not part of
+    `task_group.path`, by copying `wheel_path` into `path`.
+
+    Returns:
+        The new `wheel_path`.
+    """
+    logger = get_logger(logger_name=logger_name)
+    source = task_group.wheel_path
+    dest = (
+        Path(task_group.path) / Path(task_group.wheel_path).name
+    ).as_posix()
+    cmd = f"cp {source} {dest}"
+    logger.debug(f"[_copy_wheel_file] START {source=} {dest=}")
+    fractal_ssh.run_command(cmd=cmd)
+    logger.debug(f"[_copy_wheel_file] END {source=} {dest=}")
+    return dest

{fractal_server-2.9.0a2 → fractal_server-2.9.0a3}/fractal_server/tasks/v2/ssh/collect.py

@@ -1,5 +1,4 @@
 import logging
-import os
 import time
 from pathlib import Path
 from tempfile import TemporaryDirectory
@@ -13,16 +12,16 @@ from fractal_server.app.models.v2 import TaskGroupV2
 from fractal_server.app.schemas.v2 import TaskGroupActivityActionV2
 from fractal_server.app.schemas.v2 import TaskGroupActivityStatusV2
 from fractal_server.app.schemas.v2.manifest import ManifestV2
-from fractal_server.logger import get_logger
 from fractal_server.logger import set_logger
 from fractal_server.ssh._fabric import FractalSSH
+from fractal_server.tasks.v2.ssh._utils import _copy_wheel_file_ssh
+from fractal_server.tasks.v2.ssh._utils import _customize_and_run_template
 from fractal_server.tasks.v2.utils_background import add_commit_refresh
 from fractal_server.tasks.v2.utils_background import get_current_log
 from fractal_server.tasks.v2.utils_package_names import compare_package_names
 from fractal_server.tasks.v2.utils_python_interpreter import (
     get_python_interpreter_v2,
 )
-from fractal_server.tasks.v2.utils_templates import customize_template
 from fractal_server.tasks.v2.utils_templates import get_collection_replacements
 from fractal_server.tasks.v2.utils_templates import (
     parse_script_pip_show_stdout,
@@ -33,81 +32,6 @@ from fractal_server.utils import get_timestamp
 LOGGER_NAME = __name__
 
 
-def _customize_and_run_template(
-    *,
-    template_filename: str,
-    replacements: list[tuple[str, str]],
-    script_dir_local: str,
-    prefix: str,
-    fractal_ssh: FractalSSH,
-    script_dir_remote: str,
-    logger_name: str,
-) -> str:
-    """
-    Customize one of the template bash scripts, transfer it to the remote host
-    via SFTP and then run it via SSH.
-
-    Args:
-        template_filename: Filename of the template file (ends with ".sh").
-        replacements: Dictionary of replacements.
-        script_dir: Local folder where the script will be placed.
-        prefix: Prefix for the script filename.
-        fractal_ssh: FractalSSH object
-        script_dir_remote: Remote scripts directory
-    """
-    logger = get_logger(logger_name=logger_name)
-    logger.debug(f"_customize_and_run_template {template_filename} - START")
-
-    # Prepare name and path of script
-    if not template_filename.endswith(".sh"):
-        raise ValueError(
-            f"Invalid {template_filename=} (it must end with '.sh')."
-        )
-    script_filename = f"{prefix}_{template_filename}"
-    script_path_local = Path(script_dir_local) / script_filename
-
-    customize_template(
-        template_name=template_filename,
-        replacements=replacements,
-        script_path=script_path_local,
-    )
-
-    # Transfer script to remote host
-    script_path_remote = os.path.join(
-        script_dir_remote,
-        script_filename,
-    )
-    logger.debug(f"Now transfer {script_path_local=} over SSH.")
-    fractal_ssh.send_file(
-        local=script_path_local,
-        remote=script_path_remote,
-    )
-
-    # Execute script remotely
-    cmd = f"bash {script_path_remote}"
-    logger.debug(f"Now run '{cmd}' over SSH.")
-    stdout = fractal_ssh.run_command(cmd=cmd)
-    logger.debug(f"Standard output of '{cmd}':\n{stdout}")
-
-    logger.debug(f"_customize_and_run_template {template_filename} - END")
-    return stdout
-
-
-def _copy_wheel_file_ssh(
-    task_group: TaskGroupV2, fractal_ssh: FractalSSH
-) -> str:
-    logger = get_logger(LOGGER_NAME)
-    source = task_group.wheel_path
-    dest = (
-        Path(task_group.path) / Path(task_group.wheel_path).name
-    ).as_posix()
-    cmd = f"cp {source} {dest}"
-    logger.debug(f"[_copy_wheel_file] START {source=} {dest=}")
-    fractal_ssh.run_command(cmd=cmd)
-    logger.debug(f"[_copy_wheel_file] END {source=} {dest=}")
-    return dest
-
-
 def collect_ssh(
     *,
     task_group_id: int,
@@ -232,6 +156,7 @@ def collect_ssh(
                     new_wheel_path = _copy_wheel_file_ssh(
                         task_group=task_group,
                         fractal_ssh=fractal_ssh,
+                        logger_name=LOGGER_NAME,
                     )
                     task_group.wheel_path = new_wheel_path
                     task_group = add_commit_refresh(obj=task_group, db=db)