From 6dde40e2078be39fc8afa928bbcf8d7897628909 Mon Sep 17 00:00:00 2001
From: UmbertoJr <u.mele.coding@gmail.com>
Date: Tue, 29 Oct 2019 13:28:47 +0100
Subject: [PATCH] update

---
 code/TSP_solver.py |  2 +-
 code/__init__.py   | 10 ++++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/code/TSP_solver.py b/code/TSP_solver.py
index 2322776..d7028b6 100644
--- a/code/TSP_solver.py
+++ b/code/TSP_solver.py
@@ -32,7 +32,7 @@ class Solver_TSP:
         self.solved = True
         return self.solution
 
-    def nn(self, instance_):
+    def nn(self, instance_, starting_node=0):
         pass
 
     def plot_solution(self):
diff --git a/code/__init__.py b/code/__init__.py
index b7abdfa..80c4d39 100644
--- a/code/__init__.py
+++ b/code/__init__.py
@@ -1,2 +1,8 @@
-from code.TSP_solver import *
-from code.io_tsp import *
+import os
+
+if 'AI' in os.getcwd():
+    from code.TSP_solver import *
+    from code.io_tsp import *
+else:
+    from AI2019.code.TSP_solver import *
+    from AI2019.code.io_tsp import *
\ No newline at end of file