fixes imports

awslabs · Aug 19, 2024 · 40ae875 · 40ae875
1 parent 74aef73
commit 40ae875
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/omics/cli/run_analyzer/__init__.py b/omics/cli/run_analyzer/__init__.py
@@ -1 +1 @@
-from timeline import plot_timeline
+from . import timeline
diff --git a/omics/cli/run_analyzer/__main__.py b/omics/cli/run_analyzer/__main__.py
@@ -46,8 +46,8 @@
 import dateutil
 import dateutil.utils
 import docopt
+import timeline  # type: ignore
 from bokeh.plotting import output_file
-from timeline import plot_timeline # type: ignore
 
 exename = os.path.basename(sys.argv[0])
 OMICS_LOG_GROUP = "/aws/omics/WorkflowLog"
@@ -528,4 +528,4 @@ def tocsv(val):
         )
         title = f"arn: {run['arn']}, name: {run.get('name')}"
 
-        plot_timeline(resources, title=title, max_duration_hrs=run_duration_hrs)
+        timeline.plot_timeline(resources, title=title, max_duration_hrs=run_duration_hrs)