Extend entrypoint syntax

weigertlab · Jun 20, 2024 · 97ecefc · 97ecefc
1 parent 4ec19dc
commit 97ecefc
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/trackastra/cli.py b/trackastra/cli.py
@@ -3,9 +3,9 @@
 
 import torch
 
-from .model import Trackastra
-from .tracking.utils import graph_to_ctc, graph_to_edge_table
-from .utils import str2path
+from trackastra.model import Trackastra
+from trackastra.tracking.utils import graph_to_ctc, graph_to_edge_table
+from trackastra.utils import str2path
 
 
 def cli():
@@ -110,3 +110,7 @@ def _track_from_disk(args):
             graph=track_graph,
             outpath=outpath,
         )
+
+
+if __name__ == "__main__":
+    cli()