Merge pull request #21 from vsg-root/vsg-root-patch-2

Update Algorithm.py
vsg-root · Jul 24, 2023 · 134425b · 134425b
2 parents 7a8faa5 + 52733f0
commit 134425b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/routexplorer/algorithms/Algorithm.py b/routexplorer/algorithms/Algorithm.py
@@ -1,5 +1,11 @@
 from abc import ABC, abstractmethod
-from ..utils.Graph import Graph
+from utils.Graph import Graph
+from typing import Optional
+
+
+class InvalidReturnPoint(Exception):
+    pass
+
 
 class Algorithm(ABC):
     """
@@ -8,7 +14,7 @@ class Algorithm(ABC):
 
     @staticmethod
     @abstractmethod
-    def find_best_path(graph: Graph, show_load: bool = False, show_results: bool = False) -> tuple[list[str], int]:
+    def find_best_path(graph: Graph, return_point: Optional[str] = "R", show_load: bool = False, show_results: bool = False) -> tuple[list[str], int]:
         """
         Abstract method for finding the best path in a graph.