From 97ecefc57d9d3b6e75ddd5fbdb9b9d5e69caca27 Mon Sep 17 00:00:00 2001
From: Benjamin Gallusser <bgallusser@googlemail.com>
Date: Thu, 20 Jun 2024 12:48:31 +0200
Subject: [PATCH] Extend entrypoint syntax

---
 trackastra/cli.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/trackastra/cli.py b/trackastra/cli.py
index d9db966..db6a713 100644
--- 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()