Browse Source

Update abtest: remove print

StrayWarrior 4 days ago
parent
commit
d68c2d93d1
2 changed files with 5 additions and 4 deletions
  1. 4 3
      pqai_agent/abtest/client.py
  2. 1 1
      pqai_agent/abtest/models.py

+ 4 - 3
pqai_agent/abtest/client.py

@@ -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()

+ 1 - 1
pqai_agent/abtest/models.py

@@ -31,7 +31,7 @@ class UidDiversionBucket(DiversionBucket):
     def match(self, experiment_context):
         uid_hash = int(experiment_context.uid)
         bucket = uid_hash % self.total_buckets
-        print(f"Matching UID {experiment_context.uid} with hash {uid_hash} to bucket {bucket} in {self.buckets}")
+        # print(f"Matching UID {experiment_context.uid} with hash {uid_hash} to bucket {bucket} in {self.buckets}")
         return bucket in self.buckets