From b11cf6d07640add941d19494524f8c1af95f101c Mon Sep 17 00:00:00 2001 From: UmbertoJr Date: Sat, 9 Nov 2019 16:43:36 +0100 Subject: [PATCH] config --- src/TSP_solver.py | 4 ++++ src/__init__.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/TSP_solver.py b/src/TSP_solver.py index d305a73..c08820b 100644 --- a/src/TSP_solver.py +++ b/src/TSP_solver.py @@ -1,6 +1,10 @@ import numpy as np from matplotlib import pyplot as plt from numpy.core._multiarray_umath import ndarray +if 'AI' in os.getcwd(): + from src.utils import * +else: + from AI2019.src.utils import * diff --git a/src/__init__.py b/src/__init__.py index 175bd3e..3bc0b72 100644 --- a/src/__init__.py +++ b/src/__init__.py @@ -5,6 +5,6 @@ if 'AI' in os.getcwd(): from src.TSP_solver import * from src.io_tsp import * else: - from AI2019src.utils import * + from AI2019.src.utils import * from AI2019.src.TSP_solver import * from AI2019.src.io_tsp import *