supervisely 6.73.211__py3-none-any.whl → 6.73.212__py3-none-any.whl

supervisely/nn/inference/inference.py

@@ -1064,7 +1064,9 @@ class Inference:
             tracker = None
 
         results = []
-        batch_size = state.get("batch_size", self.get_batch_size())
+        batch_size = state.get("batch_size", None)
+        if batch_size is None:
+            batch_size = self.get_batch_size()
         tracks_data = {}
         direction = 1 if direction == "forward" else -1
         for batch in batched(
@@ -1117,7 +1119,9 @@ class Inference:
         If "output_project_id" is None, write annotations to inference request object.
         """
         logger.debug("Inferring images...", extra={"state": state})
-        batch_size = state.get("batch_size", self.get_batch_size())
+        batch_size = state.get("batch_size", None)
+        if batch_size is None:
+            batch_size = self.get_batch_size()
         output_project_id = state.get("output_project_id", None)
         images_infos = api.image.get_info_by_id_batch(images_ids)
         images_infos_dict = {im_info.id: im_info for im_info in images_infos}
@@ -1385,7 +1389,9 @@ class Inference:
             project_info = api.project.get_info_by_id(state["projectId"])
         dataset_ids = state.get("dataset_ids", None)
         cache_project_on_model = state.get("cache_project_on_model", False)
-        batch_size = state.get("batch_size", self.get_batch_size())
+        batch_size = state.get("batch_size", None)
+        if batch_size is None:
+            batch_size = self.get_batch_size()
 
         datasets_infos = api.dataset.get_list(project_info.id, recursive=True)
         if dataset_ids is not None:
@@ -2008,7 +2014,9 @@ class Inference:
             )
             images_ids = request.state.state["images_ids"]
             # check batch size
-            batch_size = request.state.state.get("batch_size", self.get_batch_size())
+            batch_size = request.state.state.get("batch_size", None)
+            if batch_size is None:
+                batch_size = self.get_batch_size()
             if self.max_batch_size is not None and batch_size > self.max_batch_size:
                 response.status_code = status.HTTP_400_BAD_REQUEST
                 return {
@@ -2045,7 +2053,9 @@ class Inference:
         def inference_video_id(response: Response, request: Request):
             logger.debug(f"'inference_video_id' request in json format:{request.state.state}")
             # check batch size
-            batch_size = request.state.state.get("batch_size", self.get_batch_size())
+            batch_size = request.state.state.get("batch_size", None)
+            if batch_size is None:
+                batch_size = self.get_batch_size()
             if self.max_batch_size is not None and batch_size > self.max_batch_size:
                 response.status_code = status.HTTP_400_BAD_REQUEST
                 return {
@@ -2131,7 +2141,9 @@ class Inference:
         def inference_video_id_async(response: Response, request: Request):
             logger.debug(f"'inference_video_id_async' request in json format:{request.state.state}")
             # check batch size
-            batch_size = request.state.state.get("batch_size", self.get_batch_size())
+            batch_size = request.state.state.get("batch_size", None)
+            if batch_size is None:
+                batch_size = self.get_batch_size()
             if self.max_batch_size is not None and batch_size > self.max_batch_size:
                 response.status_code = status.HTTP_400_BAD_REQUEST
                 return {
@@ -2173,7 +2185,9 @@ class Inference:
             if project_info.type != str(ProjectType.IMAGES):
                 raise ValueError("Only images projects are supported.")
             # check batch size
-            batch_size = request.state.state.get("batch_size", self.get_batch_size())
+            batch_size = request.state.state.get("batch_size", None)
+            if batch_size is None:
+                batch_size = self.get_batch_size()
             if self.max_batch_size is not None and batch_size > self.max_batch_size:
                 response.status_code = status.HTTP_400_BAD_REQUEST
                 return {

{supervisely-6.73.211.dist-info → supervisely-6.73.212.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.211
+Version: 6.73.212
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely

{supervisely-6.73.211.dist-info → supervisely-6.73.212.dist-info}/RECORD

@@ -777,7 +777,7 @@ supervisely/nn/benchmark/visualization/vis_metrics/reliability_diagram.py,sha256
 supervisely/nn/benchmark/visualization/vis_metrics/what_is.py,sha256=MDnYR-o7Mj-YE1Jwu9EcLUEPcu6rLknRx7LvV4nnUBo,842
 supervisely/nn/inference/__init__.py,sha256=mtEci4Puu-fRXDnGn8RP47o97rv3VTE0hjbYO34Zwqg,1622
 supervisely/nn/inference/cache.py,sha256=vjFYIkoV-txzAl_C_WKvS5odccBU8GHFY8iTxxnSqLU,25619
-supervisely/nn/inference/inference.py,sha256=7N6fMqOd18VQoTCaP4slc_cuL5ovc43FIpWomv6PtEc,114524
+supervisely/nn/inference/inference.py,sha256=pwTh6_EQ-sJNXTBD08eUMdOH8VoNF2uJUTr_P1-BuN0,114983
 supervisely/nn/inference/session.py,sha256=jmkkxbe2kH-lEgUU6Afh62jP68dxfhF5v6OGDfLU62E,35757
 supervisely/nn/inference/video_inference.py,sha256=8Bshjr6rDyLay5Za8IB8Dr6FURMO2R_v7aELasO8pR4,5746
 supervisely/nn/inference/gui/__init__.py,sha256=e3RKi93bI1r_0Dkvs_gaR1p_jkzkBMNjrcx-RVlm93k,88
@@ -957,9 +957,9 @@ supervisely/worker_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 supervisely/worker_proto/worker_api_pb2.py,sha256=VQfi5JRBHs2pFCK1snec3JECgGnua3Xjqw_-b3aFxuM,59142
 supervisely/worker_proto/worker_api_pb2_grpc.py,sha256=3BwQXOaP9qpdi0Dt9EKG--Lm8KGN0C5AgmUfRv77_Jk,28940
 supervisely_lib/__init__.py,sha256=7-3QnN8Zf0wj8NCr2oJmqoQWMKKPKTECvjH9pd2S5vY,159
-supervisely-6.73.211.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.211.dist-info/METADATA,sha256=hGxGHGtaWmL2eYrO3f6gBcacmfWyUKw4F4v8Wc_OS3w,33086
-supervisely-6.73.211.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-supervisely-6.73.211.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.211.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.211.dist-info/RECORD,,
+supervisely-6.73.212.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.212.dist-info/METADATA,sha256=iToYzdSTy2-0TOu9UbwsqvC_A_UEbTohLp4GndImxq0,33086
+supervisely-6.73.212.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+supervisely-6.73.212.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.212.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.212.dist-info/RECORD,,