|
@@ -1,6 +1,6 @@
|
|
|
# Python: experiment_client.py
|
|
|
import threading
|
|
|
-from typing import List, Dict
|
|
|
+from typing import List, Dict, Optional
|
|
|
from alibabacloud_paiabtest20240119.client import Client
|
|
|
from pqai_agent.abtest.models import Project, Domain, Layer, Experiment, ExperimentVersion, \
|
|
|
ExperimentContext, ExperimentResult
|
|
@@ -244,7 +244,7 @@ class ExperimentClient:
|
|
|
if self.running and self.worker_thread:
|
|
|
self.shutdown()
|
|
|
|
|
|
-g_client = None
|
|
|
+g_client: Optional[ExperimentClient] = None
|
|
|
|
|
|
def get_client():
|
|
|
global g_client
|
|
@@ -280,4 +280,5 @@ if __name__ == '__main__':
|
|
|
|
|
|
exp_context = ExperimentContext(uid='123')
|
|
|
result = experiment_client.match_experiment('PQAgent', exp_context)
|
|
|
- print(result)
|
|
|
+ print(result)
|
|
|
+ g_client.shutdown()
|