Skip to content

Commit

Permalink
Updating remote
Browse files Browse the repository at this point in the history
  • Loading branch information
AK2000 committed Mar 19, 2024
1 parent 009966f commit e83772b
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions compute_sdk/globus_compute_sdk/sdk/monitoring/remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,6 @@
from functools import wraps
import inspect

from parsl.multiprocessing import ForkProcess
from multiprocessing import Event, Barrier
from parsl.process_loggers import wrap_with_logs

from parsl.monitoring.message_type import MessageType
from parsl.monitoring.radios import MonitoringRadio, UDPRadio, HTEXRadio, FilesystemRadio
from typing import Any, Callable, Dict, List, Sequence, Tuple
Expand Down Expand Up @@ -36,7 +32,6 @@ def wrapped(*args: List[Any], _globus_compute_task_id: str, **kwargs: Dict[str,

task_id = _globus_compute_task_id
try_id = 0
terminate_event = Event()
# Send first message to monitoring router
send_first_message(try_id,
task_id,
Expand All @@ -56,7 +51,6 @@ def wrapped(*args: List[Any], _globus_compute_task_id: str, **kwargs: Dict[str,
return wrapped


@wrap_with_logs
def send_first_message(try_id: int,
task_id: int,
monitoring_hub_url: str,
Expand All @@ -65,7 +59,6 @@ def send_first_message(try_id: int,
radio_mode, run_dir, False)


@wrap_with_logs
def send_last_message(try_id: int,
task_id: int,
monitoring_hub_url: str,
Expand Down Expand Up @@ -106,5 +99,10 @@ def send_first_last_message(try_id: int,
'timestamp': datetime.datetime.now(),
'pid': os.getpid()
})
if not is_last:
msg[1]["task_try_time_running"] = msg[1]["timestamp"]
else:
msg[1]["task_try_time_running_ended"] = msg[1]["timestamp"]

radio.send(msg)
return

0 comments on commit e83772b

Please sign in to comment.