From d094d48b60ca8b3f4a578ca87203e8850ea00e53 Mon Sep 17 00:00:00 2001
From: JT <jcthomas000@gmail.com>
Date: Tue, 12 Sep 2023 16:24:21 +0100
Subject: [PATCH] added return type

---
 graph.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/graph.py b/graph.py
index b37ef37..2a8a28c 100644
--- a/graph.py
+++ b/graph.py
@@ -34,7 +34,7 @@ def __str__(self):
         return self.str()
 
 
-def get_graph(engine, table, primary_key):
+def get_graph(engine, table, primary_key) -> Graph:
     """Construct the graph for a specified data-point
 
     Args: