Fix

Signed-off-by: Tyler Gu <[email protected]>

performance_measurement/measure_performance.py

@@ -8,6 +8,8 @@
 from datetime import datetime
 
 import yaml
+from measure_runner import ConditionFuncType, MeasurementRunner
+from rabbitmq_inputs import RabbitMQInputGenerator
 from zk_inputs import ZooKeeperInputGenerator
 
 from acto import utils
@@ -20,9 +22,6 @@
 from acto.post_process.post_chain_inputs import ChainInputs
 from acto.utils.preprocess import process_crd
 
-from .measure_runner import ConditionFuncType, MeasurementRunner
-from .rabbitmq_inputs import RabbitMQInputGenerator
-
 sys.path.insert(0, os.path.abspath(
     os.path.join(os.path.dirname(__file__), '..')))
 

performance_measurement/measure_runner.py

@@ -7,17 +7,16 @@
 import kubernetes
 import kubernetes.client.models as k8s_models
 import yaml
+from check_utils import (check_affinity,
+                         check_persistent_volume_claim_retention_policy,
+                         check_resources, check_tolerations)
 
 from acto.common import kubernetes_client
 from acto.kubectl_client.kubectl import KubectlClient
 from acto.runner import Runner
 from acto.serialization import ActoEncoder
 from acto.utils.thread_logger import get_thread_logger
 
-from .check_utils import (check_affinity,
-                          check_persistent_volume_claim_retention_policy,
-                          check_resources, check_tolerations)
-
 ConditionFuncType = Callable[[dict, kubernetes.client.ApiClient, str], bool]