Commit 15544190 authored by Mario Teixeira Parente's avatar Mario Teixeira Parente
Browse files

adjust ARIANE imports

parent 38f97e9a
......@@ -4,8 +4,8 @@ from lib.general.benchmarking import CostMeasure
from lib.tas.benchmarking import AngleMapInducedMetric, TASAutonomousApproach, \
TASTestCase
from ariane.app.tas.server_gpr import run as run_server
from ariane.app.tas.client_stub import ClientStubTAS
from ariane.app.tas import TASClientStub
from ariane.app.tas.servers import run_gpr_server as run_ariane_server
from threading import Thread
from time import sleep
......@@ -48,7 +48,7 @@ class ARIANEApproach(TASAutonomousApproach):
scenario_name = f"{test_case.name}_{self.id}"
self.client = ClientStubTAS(intensity_function=test_case.intensity_function,
self.client = TASClientStub(intensity_function=test_case.intensity_function,
scenario_name=scenario_name, axes=test_case.axes,
offset=test_case.offset, limits=test_case.limits,
metric=lambda x1, x2: metric((x1,), (x2,)),
......@@ -70,7 +70,7 @@ class ARIANEApproach(TASAutonomousApproach):
self.client.finalize()
def _run_ariane_server(self):
run_server(num_kernel_optim_restarts=self.num_kernel_optim_restarts,
run_ariane_server(num_kernel_optim_restarts=self.num_kernel_optim_restarts,
kernel_bounds_variance=self.kernel_bounds_variance,
kernel_bounds_length_scales=self.kernel_bounds_length_scales,
num_kernel_optims_min=self.num_kernel_optims_min,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment