soorena62
20 lines
728 B
Python
20 lines
728 B
Python
import luigi
|
|
import time
|
|
from engine.tasks_writer import ImageWriter
|
|
|
|
class RadiumaPipeline(luigi.WrapperTask):
|
|
artifacts_dir = luigi.Parameter(default="artifacts")
|
|
|
|
def requires(self):
|
|
# start timer at the very beginning
|
|
self.start_time = time.time()
|
|
return ImageWriter(artifacts_dir=self.artifacts_dir)
|
|
|
|
def run(self):
|
|
elapsed = time.time() - self.start_time
|
|
hours, rem = divmod(elapsed, 3600)
|
|
minutes, seconds = divmod(rem, 60)
|
|
centiseconds = int((seconds - int(seconds)) * 100)
|
|
print(f"[Pipeline] total execution time: {int(hours):02}:{int(minutes):02}:{int(seconds):02}.{centiseconds:02}")
|
|
print("=== Workflow completed successfully ===")
|