Modified the time tracking method to accurately track time taken

ersilia/cli/commands/run.py

@@ -1,11 +1,13 @@
 import click
 import json
 import types
+import time
 
 from . import ersilia_cli
 from .. import echo
 from ... import ErsiliaModel
 from ...core.session import Session
+from ...core.tracking import RunTracker
 
 
 def run_cmd():
@@ -27,6 +29,7 @@ def run_cmd():
  help="Assume that the run is standard and, therefore, do not do so many checks.",
  )
  def run(input, output, batch_size, standard):
+ start_time = time.time()
  session = Session(config_json=None)
  model_id = session.current_model_id()
  service_class = session.current_service_class()
@@ -38,7 +41,7 @@ def run(input, output, batch_size, standard):
  fg="red",
  )
  return
-
+ 
  mdl = ErsiliaModel(
  model_id,
  service_class=service_class,
@@ -61,3 +64,17 @@ def run(input, output, batch_size, standard):
  echo("Something went wrong", fg="red")
  else:
  echo(result)
+
+ if track_runs:
+ """
+ Retrieve the time taken to run the model and update the total. 
+ """
+ time_tracker = RunTracker(
+ model_id=model_id,
+ config_json=None
+ )
+
+ time_tracker.update_total_time(
+ model_id=model_id,
+ start_time=start_time
+ ) 

ersilia/cli/commands/serve.py

@@ -1,9 +1,11 @@
 import click
+import time
 
 from .. import echo
 from . import ersilia_cli
 from ... import ErsiliaModel
 from ..messages import ModelNotFound
+from ...core.tracking import write_persistent_file
 
 
 def serve_cmd():
@@ -31,6 +33,7 @@ def serve_cmd():
  default=False,
  )
  def serve(model, lake, docker, port, track):
+ start_time = time.time()
  if docker:
  service_class = "docker"
  else:
@@ -44,6 +47,8 @@ def serve(model, lake, docker, port, track):
  )
  if not mdl.is_valid():
  ModelNotFound(mdl).echo()
+
+
  mdl.serve()
  if mdl.url is None:
  echo("No URL found. Service unsuccessful.", fg="red")
@@ -68,3 +73,12 @@ def serve(model, lake, docker, port, track):
  echo("")
  echo(":person_tipping_hand: Information:", fg="blue")
  echo(" - info", fg="blue")
+
+ if track:
+ """
+ Retrieve the time taken in seconds to serve the Model.
+ """
+ end_time = time.time()
+ duration = end_time - start_time
+ content = "Total time taken: {0}\n".format(duration)
+ write_persistent_file(content, mdl.model_id)

ersilia/core/tracking.py

@@ -16,6 +16,7 @@
 import tracemalloc
 from .session import Session
 from datetime import datetime
+from datetime import timedelta
 from .base import ErsiliaBase
 from collections import defaultdict
 from ..default import EOS, ERSILIA_RUNS_FOLDER
@@ -387,6 +388,50 @@ def __init__(self, model_id, config_json):
 
 
 # return stats
+
+
+ def update_total_time(self, model_id, start_time):
+ """
+ Method to track and update the Total time taken by model.
+ :Param model_id: The currently running model.
+ :Param start_time: The start time of the running model. 
+ """
+
+ end_time = time.time()
+ duration = end_time - start_time
+ if check_file_exists(model_id):
+ file_name = get_persistent_file_path(model_id)
+ with open(file_name, "r") as f:
+ lines = f.readlines()
+
+ updated_lines = []
+ total_time_found = False
+
+ for line in lines:
+ if "Total time taken" in line and not total_time_found:
+ try:
+ total_time_str = line.split(":")[1].strip()
+ total_time = float(total_time_str)
+ total_time += duration
+ formatted_time = str(timedelta(seconds=total_time))
+ updated_lines.append(f"Total time taken: {formatted_time}\n")
+ total_time_found = True
+ except (ValueError, IndexError) as e:
+ print(f"Error parsing 'Total time taken' value: {e}")
+ else:
+ updated_lines.append(line)
+
+ if not total_time_found:
+ updated_lines.append(f"Total time taken: {formatted_duration}\n")
+
+ new_content = "".join(updated_lines)
+ with open(file_name, "w") as f:
+ f.write(f"{new_content}\n")
+ else:
+ new_content = f"Total time: {formatted_duration}\n"
+ with open(file_name, "w") as f:
+ f.write(f"{new_content}\n")
+
 
  def get_file_sizes(self, input_file, output_file):
  """
@@ -533,7 +578,7 @@ def track(self, input, result, meta):
  """
  Tracks the results of a model run.
  """
- self.time_start = datetime.now()
+
  self.docker_client = SimpleDocker()
  json_dict = {}
  input_data = read_csv(input)
@@ -563,9 +608,6 @@ def track(self, input, result, meta):
  model_id = meta["metadata"].get("Identifier", "Unknown")
  json_dict["model_id"] = model_id
 
- time_taken = datetime.now() - self.time_start
- json_dict["time_taken in seconds"] = str(time_taken)
-
  # checking for mismatched types
  nan_count = get_nan_counts(result_data)
  json_dict["nan_count"] = nan_count
@@ -600,4 +642,4 @@ def track(self, input, result, meta):
  def log(self, result, meta):
  self.log_result(result)
  self.log_meta(meta)
- self.log_logs()
+ self.log_logs()