feat generate 升级 attribute retrieve 迁移

app/service/super_resolution/service.py

@@ -24,11 +24,7 @@ class SuperResolution:
         self.user_id = self.tasks_id[self.tasks_id.rfind('-') + 1:]
         self.sr_image_url = data.sr_image_url
         self.sr_xn = data.sr_xn
-        self.minio_client = Minio(
-            f"{MINIO_IP}:{MINIO_PORT}",
-            access_key=MINIO_ACCESS,
-            secret_key=MINIO_SECRET,
-            secure=MINIO_SECURE)
+        self.minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
         self.redis_client.set(self.tasks_id, json.dumps({'status': 'PENDING', 'message': "pending", 'data': ''}))
         self.connection = pika.BlockingConnection(pika.ConnectionParameters(**RABBITMQ_PARAMS))
         self.channel = self.connection.channel()
@@ -38,7 +34,7 @@ class SuperResolution:
         self.triton_client.close()
         self.connection.close()
 
-    @RunTime
+    # @RunTime
     def read_image(self):
         try:
             image_data = self.minio_client.get_object(self.sr_image_url.split("/", 1)[0], self.sr_image_url.split("/", 1)[1])
@@ -56,7 +52,7 @@ class SuperResolution:
         logging.info(f"{self.tasks_id} ===> {status_data}")
         return status_data
 
-    @RunTime
+    # @RunTime
     def infer(self, inputs):
         return self.triton_client.async_infer(
             model_name=SR_MODEL_NAME,
@@ -64,7 +60,7 @@ class SuperResolution:
             callback=self.callback
         )
 
-    @RunTime
+    # @RunTime
     def sr_result(self):
         sample = self.read_image()
         if self.sr_xn == 2:

app/service/super_resolution/test.py

@@ -1,2 +0,0 @@
-a = "123-86"
-print(a[a.rfind('-') + 1:])