Skip to content

Commit a8ccb10

Browse files
committed
moved import statements within if condition to ensure that script doesn't fail
1 parent 77c8c70 commit a8ccb10

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

algorithms/optimize.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
11
# Script for optimizing a given problem using a specific method, Note: this assumes maximization problem
22

33
import argparse, os
4-
from algorithms import *
5-
from DyCors import minimize
64
from test_problems.problem import TestProblem
75
import numpy as np
86
from scipy.io import savemat
97
from time import time
8+
from utils import set_seed, latin_hypercube
109

1110
parser = argparse.ArgumentParser(description='Runscript for running BO algorithm')
1211

@@ -66,6 +65,8 @@
6665

6766
if method == "bo":
6867

68+
from algorithms.bo import BO
69+
6970
optimizer = BO(
7071
f=f,
7172
lb=f.lb,
@@ -76,6 +77,8 @@
7677

7778
elif method == "ibnn":
7879

80+
from algorithms.ibnn import IBNN
81+
7982
optimizer = IBNN(
8083
f=f,
8184
lb=f.lb,
@@ -86,6 +89,8 @@
8689

8790
elif method == "turbo":
8891

92+
from turbo import Turbo
93+
8994
optimizer = Turbo(
9095
f=f,
9196
lb=f.lb,
@@ -96,6 +101,8 @@
96101

97102
elif method == "dycors":
98103

104+
from DyCors import minimize
105+
99106
# options for dycors - from original source code
100107
options = {
101108
"Nmax": max_evals,
@@ -116,6 +123,8 @@
116123

117124
elif method == "snbo":
118125

126+
from snbo import SNBO
127+
119128
optimizer = SNBO(
120129
f=f,
121130
lb=f.lb,

0 commit comments

Comments
 (0)