Update basic_graphs.py

Author: itsvinayak

graphs/basic_graphs.py

@@ -220,7 +220,7 @@ def prim(G, s):
                 if v[1] < dist.get(v[0], 100000):
                     dist[v[0]] = v[1]
                     path[v[0]] = u
-    return sum(dist.values())
+    return dist
 
 
 """