From 6dde40e2078be39fc8afa928bbcf8d7897628909 Mon Sep 17 00:00:00 2001 From: UmbertoJr 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