Skip to content

Commit

Permalink
code cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
henricasanova committed Sep 14, 2024
1 parent eb1d56a commit a74a0c2
Show file tree
Hide file tree
Showing 11 changed files with 11 additions and 9 deletions.
2 changes: 2 additions & 0 deletions wrench/action.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@
# (at your option) any later version.
from wrench.simulation_item import SimulationItem
from wrench.compound_job import CompoundJob


# from enum import Enum


Expand Down
2 changes: 1 addition & 1 deletion wrench/batch_compute_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,4 +51,4 @@ def submit_compound_job(self, compound_job: CompoundJob, service_specific_args:
:param service_specific_args: the service-specific arguments
:type service_specific_args: dict[str, str]
"""
return self.simulation._submit_compound_job(compound_job, self, json.dumps(service_specific_args))
return self.simulation._submit_compound_job(compound_job, self, json.dumps(service_specific_args))
2 changes: 1 addition & 1 deletion wrench/cloud_compute_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,4 +53,4 @@ def destroy_vm(self, vm: VirtualMachine) -> None:
:param vm: A virtual machine
:type vm: VirtualMachine
"""
return self.simulation._destroy_vm(vm)
return self.simulation._destroy_vm(vm)
1 change: 0 additions & 1 deletion wrench/compound_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,4 +183,3 @@ def __repr__(self) -> str:
"""
s = f"CompoundJob(name={self.name})"
return s

3 changes: 2 additions & 1 deletion wrench/compute_action.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,13 +11,14 @@

from typing import Tuple


class ComputeAction(Action):
"""
WRENCH Action class
"""

def __init__(self, simulation, compound_job: CompoundJob, name: str, flops: float, ram: float, min_num_cores: int,
max_num_cores: int, parallel_model: tuple) -> None:

"""
Constructor
:param simulation: simulation object
Expand Down
1 change: 0 additions & 1 deletion wrench/file_copy_action.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ class FileCopyAction(Action):

def __init__(self, simulation, compound_job: CompoundJob, name: str, file: File,
src_storage_service: StorageService, dest_storage_service: StorageService, uses_scratch: bool) -> None:

"""
Constructor
:param simulation: simulation object
Expand Down
2 changes: 1 addition & 1 deletion wrench/file_delete_action.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,9 @@ class FileDeleteAction(Action):
"""
WRENCH File Delete Action class
"""

def __init__(self, simulation, compound_job: CompoundJob, name: str, file: File,
storage_service: StorageService, uses_scratch: bool) -> None:

"""
Constructor
:param simulation: simulation object
Expand Down
2 changes: 1 addition & 1 deletion wrench/file_registry_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ class FileRegistryService(SimulationItem):
"""
WRENCH File Registry Service class
"""

def __init__(self, simulation, name: str) -> None:
"""
Constructor
Expand Down Expand Up @@ -77,4 +78,3 @@ def __repr__(self) -> str:
"""
s = f"FileRegistryService(name={self.name})"
return s

2 changes: 1 addition & 1 deletion wrench/file_write_action.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,9 @@ class FileWriteAction(Action):
"""
WRENCH File Write Action class
"""

def __init__(self, simulation, compound_job: CompoundJob, name: str, file: File,
storage_service: StorageService, uses_scratch: bool) -> None:

"""
Constructor
:param simulation: simulation object
Expand Down
1 change: 1 addition & 0 deletions wrench/sleep_action.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ class SleepAction(Action):
"""
WRENCH Sleep Action class
"""

def __init__(self, simulation, compound_job: CompoundJob, name: str, sleep_time: float) -> None:
"""
Constructor
Expand Down
2 changes: 1 addition & 1 deletion wrench/workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

from typing import TYPE_CHECKING

if TYPE_CHECKING: # pragma: no cover
if TYPE_CHECKING: # pragma: no cover
from wrench.simulation import Simulation
from wrench.task import Task
from wrench.file import File
Expand Down

0 comments on commit a74a0c2

Please sign in to comment.