fractal-server 2.17.1a0__py3-none-any.whl → 2.17.2__py3-none-any.whl

fractal_server/runner/executors/slurm_sudo/runner.py

@@ -5,15 +5,20 @@ import subprocess  # nosec
 from concurrent.futures import ThreadPoolExecutor
 from pathlib import Path
 
-from ..slurm_common.base_slurm_runner import BaseSlurmRunner
-from ..slurm_common.slurm_job_task_models import SlurmJob
-from ._subprocess_run_as_user import _mkdir_as_user
-from ._subprocess_run_as_user import _run_command_as_user
 from fractal_server.app.models import Profile
 from fractal_server.app.models import Resource
 from fractal_server.logger import set_logger
 from fractal_server.runner.config import JobRunnerConfigSLURM
 from fractal_server.runner.exceptions import JobExecutionError
+from fractal_server.runner.executors.slurm_common.base_slurm_runner import (
+    BaseSlurmRunner,
+)
+from fractal_server.runner.executors.slurm_common.slurm_job_task_models import (
+    SlurmJob,
+)
+
+from ._subprocess_run_as_user import _mkdir_as_user
+from ._subprocess_run_as_user import _run_command_as_user
 
 logger = set_logger(__name__)
 
@@ -40,7 +45,11 @@ def _subprocess_run_or_raise(
         raise JobExecutionError(info=error_msg)
 
 
-class SudoSlurmRunner(BaseSlurmRunner):
+class SlurmSudoRunner(BaseSlurmRunner):
+    """
+    Runner implementation for a computational `sudo_slurm` resource.
+    """
+
     slurm_user: str
     slurm_account: str | None = None
 
@@ -63,9 +72,7 @@ class SudoSlurmRunner(BaseSlurmRunner):
         """
 
         self.slurm_user = profile.username
-        self.shared_config = JobRunnerConfigSLURM(
-            **resource.jobs_runner_config
-        )
+        self.shared_config = JobRunnerConfigSLURM(**resource.jobs_runner_config)
 
         super().__init__(
             slurm_runner_type="sudo",
@@ -90,9 +97,7 @@ class SudoSlurmRunner(BaseSlurmRunner):
         """
         Fetch artifacts for a single SLURM jobs.
         """
-        logger.debug(
-            f"[_fetch_artifacts_single_job] {job.slurm_job_id=} START"
-        )
+        logger.debug(f"[_fetch_artifacts_single_job] {job.slurm_job_id=} START")
         source_target_list = [
             (job.slurm_stdout_remote, job.slurm_stdout_local),
             (job.slurm_stderr_remote, job.slurm_stderr_local),

fractal_server/runner/task_files.py

@@ -86,9 +86,7 @@ class TaskFiles(BaseModel):
     @property
     def log_file_remote_path(self) -> Path:
         self._check_component()
-        return (
-            self.wftask_subfolder_remote / f"{self.prefix_component}-log.txt"
-        )
+        return self.wftask_subfolder_remote / f"{self.prefix_component}-log.txt"
 
     @property
     def log_file_remote(self) -> str:
@@ -141,8 +139,16 @@ def enrich_task_files_multisubmit(
 ) -> list[TaskFiles]:
     """
     Expand `TaskFiles` objects with `component` and `prefix`.
+
+    Args:
+        tot_tasks: Total number of images to process.
+        batch_size: Batch size, where `0` means `batch_size=tot_tasks`.
+        base_task_files: Original `TaskFiles` object to be enriched.
     """
 
+    # Replace `batch_size=0` with `batch_size=tot_tasks`
+    batch_size = batch_size or tot_tasks
+
     new_list_task_files: list[TaskFiles] = []
     for absolute_index in range(tot_tasks):
         ind_batch = absolute_index // batch_size

fractal_server/runner/v2/_local.py

@@ -1,16 +1,21 @@
 from pathlib import Path
 
-from ..executors.local.get_local_config import get_local_backend_config
-from ..executors.local.runner import LocalRunner
-from ..set_start_and_last_task_index import set_start_and_last_task_index
-from .runner import execute_tasks_v2
 from fractal_server.app.models.v2 import DatasetV2
 from fractal_server.app.models.v2 import Profile
 from fractal_server.app.models.v2 import Resource
 from fractal_server.app.models.v2 import WorkflowV2
+from fractal_server.runner.executors.local.get_local_config import (
+    get_local_backend_config,
+)
+from fractal_server.runner.executors.local.runner import LocalRunner
+from fractal_server.runner.set_start_and_last_task_index import (
+    set_start_and_last_task_index,
+)
 from fractal_server.ssh._fabric import FractalSSH
 from fractal_server.types import AttributeFilters
 
+from .runner import execute_tasks_v2
+
 
 def process_workflow(
     *,

fractal_server/runner/v2/_slurm_ssh.py

@@ -15,20 +15,26 @@ Slurm Backend
 
 This backend runs fractal workflows in a SLURM cluster.
 """
+
 from pathlib import Path
 
-from ...ssh._fabric import FractalSSH
-from ..executors.slurm_common.get_slurm_config import get_slurm_config
-from ..executors.slurm_ssh.runner import SlurmSSHRunner
-from ..set_start_and_last_task_index import set_start_and_last_task_index
-from .runner import execute_tasks_v2
 from fractal_server.app.models.v2 import DatasetV2
 from fractal_server.app.models.v2 import Profile
 from fractal_server.app.models.v2 import Resource
 from fractal_server.app.models.v2 import WorkflowV2
 from fractal_server.logger import set_logger
+from fractal_server.runner.executors.slurm_common.get_slurm_config import (
+    get_slurm_config,
+)
+from fractal_server.runner.executors.slurm_ssh.runner import SlurmSSHRunner
+from fractal_server.runner.set_start_and_last_task_index import (
+    set_start_and_last_task_index,
+)
+from fractal_server.ssh._fabric import FractalSSH
 from fractal_server.types import AttributeFilters
 
+from .runner import execute_tasks_v2
+
 logger = set_logger(__name__)
 
 

fractal_server/runner/v2/_slurm_sudo.py

@@ -15,19 +15,25 @@ Slurm Backend
 
 This backend runs fractal workflows in a SLURM cluster.
 """
+
 from pathlib import Path
 
-from ..executors.slurm_common.get_slurm_config import get_slurm_config
-from ..executors.slurm_sudo.runner import SudoSlurmRunner
-from ..set_start_and_last_task_index import set_start_and_last_task_index
-from .runner import execute_tasks_v2
 from fractal_server.app.models.v2 import DatasetV2
 from fractal_server.app.models.v2 import Profile
 from fractal_server.app.models.v2 import Resource
 from fractal_server.app.models.v2 import WorkflowV2
+from fractal_server.runner.executors.slurm_common.get_slurm_config import (
+    get_slurm_config,
+)
+from fractal_server.runner.executors.slurm_sudo.runner import SlurmSudoRunner
+from fractal_server.runner.set_start_and_last_task_index import (
+    set_start_and_last_task_index,
+)
 from fractal_server.ssh._fabric import FractalSSH
 from fractal_server.types import AttributeFilters
 
+from .runner import execute_tasks_v2
+
 
 def process_workflow(
     *,
@@ -94,7 +100,7 @@ def process_workflow(
     if isinstance(worker_init, str):
         worker_init = worker_init.split("\n")
 
-    with SudoSlurmRunner(
+    with SlurmSudoRunner(
         root_dir_local=workflow_dir_local,
         root_dir_remote=workflow_dir_remote,
         common_script_lines=worker_init,

fractal_server/runner/v2/db_tools.py

@@ -10,7 +10,6 @@ from fractal_server.app.models.v2 import HistoryUnit
 from fractal_server.app.schemas.v2 import HistoryUnitStatus
 from fractal_server.logger import set_logger
 
-
 _CHUNK_SIZE = 2_000
 
 logger = set_logger(__name__)

fractal_server/runner/v2/deduplicate_list.py

@@ -1,9 +1,10 @@
 from typing import TypeVar
 
-from .task_interface import InitArgsModel
 from fractal_server.images import SingleImage
 from fractal_server.images import SingleImageTaskOutput
 
+from .task_interface import InitArgsModel
+
 T = TypeVar("T", SingleImage, SingleImageTaskOutput, InitArgsModel)
 
 

fractal_server/runner/v2/runner.py

@@ -7,13 +7,6 @@ from sqlalchemy.orm.attributes import flag_modified
 from sqlmodel import delete
 from sqlmodel import update
 
-from .merge_outputs import merge_outputs
-from .runner_functions import GetRunnerConfigType
-from .runner_functions import run_v2_task_compound
-from .runner_functions import run_v2_task_non_parallel
-from .runner_functions import run_v2_task_parallel
-from .runner_functions import SubmissionOutcome
-from .task_interface import TaskOutput
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models.v2 import AccountingRecord
 from fractal_server.app.models.v2 import DatasetV2
@@ -28,8 +21,8 @@ from fractal_server.app.schemas.v2 import TaskDumpV2
 from fractal_server.app.schemas.v2 import TaskGroupDumpV2
 from fractal_server.app.schemas.v2 import TaskType
 from fractal_server.images import SingleImage
-from fractal_server.images.status_tools import enrich_images_unsorted_sync
 from fractal_server.images.status_tools import IMAGE_STATUS_KEY
+from fractal_server.images.status_tools import enrich_images_unsorted_sync
 from fractal_server.images.tools import filter_image_list
 from fractal_server.images.tools import find_image_by_zarr_url
 from fractal_server.images.tools import merge_type_filters
@@ -39,6 +32,14 @@ from fractal_server.runner.executors.base_runner import BaseRunner
 from fractal_server.runner.v2.db_tools import update_status_of_history_run
 from fractal_server.types import AttributeFilters
 
+from .merge_outputs import merge_outputs
+from .runner_functions import GetRunnerConfigType
+from .runner_functions import SubmissionOutcome
+from .runner_functions import run_v2_task_compound
+from .runner_functions import run_v2_task_non_parallel
+from .runner_functions import run_v2_task_parallel
+from .task_interface import TaskOutput
+
 
 def _remove_status_from_attributes(
     images: list[dict[str, Any]],
@@ -312,9 +313,7 @@ def execute_tasks_v2(
             # history status.
             for image_obj in current_task_output.image_list_updates:
                 image = image_obj.model_dump()
-                if image["zarr_url"] in [
-                    img["zarr_url"] for img in tmp_images
-                ]:
+                if image["zarr_url"] in [img["zarr_url"] for img in tmp_images]:
                     img_search = find_image_by_zarr_url(
                         images=tmp_images,
                         zarr_url=image["zarr_url"],
@@ -418,9 +417,7 @@ def execute_tasks_v2(
 
             # Update type_filters based on task-manifest output_types
             type_filters_from_task_manifest = task.output_types
-            current_dataset_type_filters.update(
-                type_filters_from_task_manifest
-            )
+            current_dataset_type_filters.update(type_filters_from_task_manifest)
         except Exception as e:
             logger.error(
                 "Unexpected error in post-task-execution block. "

fractal_server/runner/v2/runner_functions.py

@@ -6,12 +6,6 @@ from typing import Protocol
 from pydantic import BaseModel
 from pydantic import ConfigDict
 
-from ..exceptions import JobExecutionError
-from ..exceptions import TaskOutputValidationError
-from .db_tools import update_status_of_history_unit
-from .deduplicate_list import deduplicate_list
-from .task_interface import InitTaskOutput
-from .task_interface import TaskOutput
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models.v2 import HistoryUnit
 from fractal_server.app.models.v2 import TaskV2
@@ -22,16 +16,16 @@ from fractal_server.exceptions import UnreachableBranchError
 from fractal_server.logger import set_logger
 from fractal_server.runner.config import JobRunnerConfigLocal
 from fractal_server.runner.config import JobRunnerConfigSLURM
+from fractal_server.runner.exceptions import JobExecutionError
+from fractal_server.runner.exceptions import TaskOutputValidationError
 from fractal_server.runner.executors.base_runner import BaseRunner
 from fractal_server.runner.executors.slurm_common.slurm_config import (
     SlurmConfig,
 )
-from fractal_server.runner.task_files import enrich_task_files_multisubmit
 from fractal_server.runner.task_files import SUBMIT_PREFIX
 from fractal_server.runner.task_files import TaskFiles
-from fractal_server.runner.v2.db_tools import (
-    bulk_update_status_of_history_unit,
-)
+from fractal_server.runner.task_files import enrich_task_files_multisubmit
+from fractal_server.runner.v2.db_tools import bulk_update_status_of_history_unit
 from fractal_server.runner.v2.db_tools import bulk_upsert_image_cache_fast
 from fractal_server.runner.v2.task_interface import (
     _cast_and_validate_InitTaskOutput,
@@ -40,6 +34,11 @@ from fractal_server.runner.v2.task_interface import (
     _cast_and_validate_TaskOutput,
 )
 
+from .db_tools import update_status_of_history_unit
+from .deduplicate_list import deduplicate_list
+from .task_interface import InitTaskOutput
+from .task_interface import TaskOutput
+
 
 class GetRunnerConfigTypeLocal(Protocol):
     def __call__(
@@ -48,8 +47,7 @@ class GetRunnerConfigTypeLocal(Protocol):
         wftask: WorkflowTaskV2,
         which_type: Literal["non_parallel", "parallel"],
         tot_tasks: int,
-    ) -> JobRunnerConfigLocal:
-        ...
+    ) -> JobRunnerConfigLocal: ...
 
 
 class GetRunnerConfigTypeSLURM(Protocol):
@@ -59,8 +57,7 @@ class GetRunnerConfigTypeSLURM(Protocol):
         wftask: WorkflowTaskV2,
         which_type: Literal["non_parallel", "parallel"],
         tot_tasks: int,
-    ) -> SlurmConfig:
-        ...
+    ) -> SlurmConfig: ...
 
 
 GetRunnerConfigType = GetRunnerConfigTypeLocal | GetRunnerConfigTypeSLURM

fractal_server/runner/v2/submit_workflow.py

@@ -5,6 +5,7 @@ This module is the single entry point to the runner backend subsystem V2.
 Other subsystems should only import this module and not its submodules or
 the individual backends.
 """
+
 import os
 import traceback
 from pathlib import Path
@@ -12,9 +13,6 @@ from typing import Protocol
 
 from sqlalchemy.orm import Session as DBSyncSession
 
-from ._local import process_workflow as local_process_workflow
-from ._slurm_ssh import process_workflow as slurm_ssh_process_workflow
-from ._slurm_sudo import process_workflow as slurm_sudo_process_workflow
 from fractal_server import __VERSION__
 from fractal_server.app.db import DB
 from fractal_server.app.models.v2 import DatasetV2
@@ -34,6 +32,10 @@ from fractal_server.types import AttributeFilters
 from fractal_server.utils import get_timestamp
 from fractal_server.zip_tools import _zip_folder_to_file_and_remove
 
+from ._local import process_workflow as local_process_workflow
+from ._slurm_ssh import process_workflow as slurm_ssh_process_workflow
+from ._slurm_sudo import process_workflow as slurm_sudo_process_workflow
+
 
 class ProcessWorkflowType(Protocol):
     def __call__(
@@ -53,8 +55,7 @@ class ProcessWorkflowType(Protocol):
         resource: Resource,
         profile: Profile,
         user_cache_dir: str,
-    ) -> None:
-        ...
+    ) -> None: ...
 
 
 def fail_job(
@@ -119,7 +120,7 @@ def submit_workflow(
             Computational resource to be used for this job (e.g. a SLURM
             cluster).
         profile:
-           Computational profile to be used for this job.
+            Computational profile to be used for this job.
         fractal_ssh: SSH object, for when `resource.type = "slurm_ssh"`.
     """
     # Declare runner backend and set `process_workflow` function

fractal_server/ssh/_fabric.py

@@ -16,12 +16,11 @@ from invoke import UnexpectedExit
 from paramiko.ssh_exception import NoValidConnectionsError
 from pydantic import BaseModel
 
-from ..logger import close_logger
-from ..logger import get_logger
-from ..logger import set_logger
+from fractal_server.logger import close_logger
+from fractal_server.logger import get_logger
+from fractal_server.logger import set_logger
 from fractal_server.string_tools import validate_cmd
 
-
 SSH_MONITORING_LOGGER_NAME = "ssh-log"
 
 
@@ -99,8 +98,7 @@ def _acquire_lock_with_timeout(
         if not result:
             logger.error(f"Lock for '{label}' was *not* acquired.")
             raise FractalSSHTimeoutError(
-                f"Failed to acquire lock for '{label}' within "
-                f"{timeout} seconds"
+                f"Failed to acquire lock for '{label}' within {timeout} seconds"
             )
         t_lock_acquisition = time.perf_counter()
         elapsed = t_lock_acquisition - t_lock_request
@@ -374,8 +372,6 @@ class FractalSSH:
         Args:
             cmd: Command to be run
             allow_char: Forbidden chars to allow for this command
-            max_attempts:
-            base_interval:
             lock_timeout:
 
         Returns:
@@ -458,9 +454,7 @@ class FractalSSH:
                 logger_name=self.logger_name,
             ):
                 self._sftp_unsafe().put(local, remote)
-            self.logger.info(
-                f"[send_file] END transfer of '{local}' over SSH."
-            )
+            self.logger.info(f"[send_file] END transfer of '{local}' over SSH.")
         except Exception as e:
             self.log_and_raise(
                 e=e,
@@ -766,8 +760,7 @@ class FractalSSHList:
             logger_name=self._logger_name,
         ):
             self.logger.info(
-                f"Removing FractalSSH object for {user}@{host} "
-                "from collection."
+                f"Removing FractalSSH object for {user}@{host} from collection."
             )
             fractal_ssh_obj = self._data.pop(key)
             self.logger.info(

fractal_server/string_tools.py

@@ -1,6 +1,5 @@
 import string
 
-
 __SPECIAL_CHARACTERS__ = f"{string.punctuation}{string.whitespace}"
 
 # List of invalid characters discussed here:

fractal_server/syringe.py

@@ -34,11 +34,11 @@ or popped from the directory.
     >>> bar()
     42
 """
+
 from collections.abc import Callable
 from typing import Any
 from typing import TypeVar
 
-
 T = TypeVar("T")
 _instance_count = 0
 

fractal_server/tasks/config/_pixi.py

@@ -3,8 +3,8 @@ from typing import Annotated
 
 from pydantic import AfterValidator
 from pydantic import BaseModel
-from pydantic import model_validator
 from pydantic import PositiveInt
+from pydantic import model_validator
 
 from fractal_server.types import DictStrStr
 from fractal_server.types import NonEmptyStr

fractal_server/tasks/config/_python.py

@@ -14,12 +14,25 @@ class TasksPythonSettings(BaseModel):
 
     For task collection to work, there must be one or more base Python
     interpreters available on your system.
+
+    Attributes:
+        default_version:
+            Default task-collection Python version (must be a key of
+            `versions`).
+        versions:
+            Dictionary mapping Python versions to the corresponding
+            interpreters. Example:
+            ```json
+            {
+              "3.11": "/path/to/python3.11",
+              "3.13": "/path/to/python3.13"
+            }
+            ```
+        pip_cache_dir:
+            Argument for `--cache-dir` option of `pip install`, if set.
     """
 
     default_version: NonEmptyStr
-    """
-    Default task-collection Python version (must be a key of `versions`).
-    """
     versions: dict[
         Literal[
             "3.9",
@@ -31,14 +44,8 @@ class TasksPythonSettings(BaseModel):
         ],
         AbsolutePathStr,
     ]
-    """
-    Dictionary mapping Python versions to the corresponding interpreters.
-    """
 
     pip_cache_dir: AbsolutePathStr | None = None
-    """
-    Argument for `--cache-dir` option of `pip install`, if set.
-    """
 
     @model_validator(mode="after")
     def _validate_versions(self) -> Self:

fractal_server/tasks/utils.py

@@ -1,6 +1,5 @@
 from pathlib import Path
 
-
 COLLECTION_FILENAME = "collection.json"
 COLLECTION_LOG_FILENAME = "collection.log"
 COLLECTION_FREEZE_FILENAME = "collection_freeze.txt"

fractal_server/tasks/v2/local/_utils.py

@@ -1,10 +1,10 @@
 from pathlib import Path
 
-from ..utils_pixi import simplify_pyproject_toml
 from fractal_server.app.models import Resource
 from fractal_server.app.schemas.v2 import TaskCreateV2
 from fractal_server.logger import get_logger
 from fractal_server.logger import set_logger
+from fractal_server.tasks.v2.utils_pixi import simplify_pyproject_toml
 from fractal_server.tasks.v2.utils_templates import customize_template
 from fractal_server.utils import execute_command_sync
 

fractal_server/tasks/v2/local/collect.py

@@ -4,8 +4,6 @@ import time
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
-from ..utils_database import create_db_tasks_and_update_task_group_sync
-from ._utils import _customize_and_run_template
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models import Profile
 from fractal_server.app.models import Resource
@@ -20,22 +18,23 @@ 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 add_commit_refresh
 from fractal_server.tasks.v2.utils_background import fail_and_cleanup
-from fractal_server.tasks.v2.utils_background import (
-    get_activity_and_task_group,
-)
+from fractal_server.tasks.v2.utils_background import get_activity_and_task_group
 from fractal_server.tasks.v2.utils_background import get_current_log
 from fractal_server.tasks.v2.utils_background import prepare_tasks_metadata
+from fractal_server.tasks.v2.utils_database import (
+    create_db_tasks_and_update_task_group_sync,
+)
 from fractal_server.tasks.v2.utils_package_names import compare_package_names
 from fractal_server.tasks.v2.utils_python_interpreter import (
     get_python_interpreter,
 )
-from fractal_server.tasks.v2.utils_templates import get_collection_replacements
-from fractal_server.tasks.v2.utils_templates import (
-    parse_script_pip_show_stdout,
-)
 from fractal_server.tasks.v2.utils_templates import SCRIPTS_SUBFOLDER
+from fractal_server.tasks.v2.utils_templates import get_collection_replacements
+from fractal_server.tasks.v2.utils_templates import parse_script_pip_show_stdout
 from fractal_server.utils import get_timestamp
 
+from ._utils import _customize_and_run_template
+
 
 def collect_local(
     *,
@@ -234,7 +233,7 @@ def collect_local(
                 activity.log = get_current_log(log_file_path)
                 activity = add_commit_refresh(obj=activity, db=db)
 
-                logger.info("create_db_tasks_and_update_task_group - " "start")
+                logger.info("create_db_tasks_and_update_task_group - start")
                 create_db_tasks_and_update_task_group_sync(
                     task_list=task_list,
                     task_group_id=task_group.id,
@@ -274,8 +273,7 @@ def collect_local(
                     logger.info(f"Deleted folder {task_group.path}")
                 except Exception as rm_e:
                     logger.error(
-                        "Removing folder failed.\n"
-                        f"Original error:\n{str(rm_e)}"
+                        f"Removing folder failed.\nOriginal error:\n{str(rm_e)}"
                     )
 
                 fail_and_cleanup(

fractal_server/tasks/v2/local/collect_pixi.py

@@ -4,10 +4,6 @@ import time
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
-from ..utils_database import create_db_tasks_and_update_task_group_sync
-from ..utils_pixi import parse_collect_stdout
-from ..utils_pixi import SOURCE_DIR_NAME
-from ._utils import edit_pyproject_toml_in_place_local
 from fractal_server.app.db import get_sync_db
 from fractal_server.app.models import Profile
 from fractal_server.app.models import Resource
@@ -22,15 +18,20 @@ from fractal_server.tasks.v2.local._utils import _customize_and_run_template
 from fractal_server.tasks.v2.local._utils import check_task_files_exist
 from fractal_server.tasks.v2.utils_background import add_commit_refresh
 from fractal_server.tasks.v2.utils_background import fail_and_cleanup
-from fractal_server.tasks.v2.utils_background import (
-    get_activity_and_task_group,
-)
+from fractal_server.tasks.v2.utils_background import get_activity_and_task_group
 from fractal_server.tasks.v2.utils_background import get_current_log
 from fractal_server.tasks.v2.utils_background import prepare_tasks_metadata
+from fractal_server.tasks.v2.utils_database import (
+    create_db_tasks_and_update_task_group_sync,
+)
+from fractal_server.tasks.v2.utils_pixi import SOURCE_DIR_NAME
+from fractal_server.tasks.v2.utils_pixi import parse_collect_stdout
 from fractal_server.tasks.v2.utils_templates import SCRIPTS_SUBFOLDER
 from fractal_server.utils import execute_command_sync
 from fractal_server.utils import get_timestamp
 
+from ._utils import edit_pyproject_toml_in_place_local
+
 
 def collect_local_pixi(
     *,
@@ -177,9 +178,7 @@ def collect_local_pixi(
                 package_root = parsed_output["package_root"]
                 venv_size = parsed_output["venv_size"]
                 venv_file_number = parsed_output["venv_file_number"]
-                project_python_wrapper = parsed_output[
-                    "project_python_wrapper"
-                ]
+                project_python_wrapper = parsed_output["project_python_wrapper"]
 
                 # Make task folder 755
                 command = f"chmod -R 755 {source_dir}"