neuromeka-vfm 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

neuromeka_vfm/segmentation.py

@@ -26,6 +26,25 @@ class Segmentation:
             self.call_time = {"add_image_prompt": 0, "register_first_frame": 0, "get_next": 0}
             self.call_count = {"add_image_prompt": 0, "register_first_frame": 0, "get_next": 0}
 
+    def _is_success(self, response):
+        """
+        Normalize server success flag.
+        Some servers return {"result": "SUCCESS"}, others {"success": true}, and
+        the segmentation server returns {"status": "success"}.
+        """
+        # Try known keys in order of common usage.
+        for key in ("result", "success", "status"):
+            flag = response.get(key)
+            if flag is None:
+                continue
+            if isinstance(flag, str):
+                return flag.lower() == "success"
+            if isinstance(flag, bool):
+                return flag
+            return bool(flag)
+        # Fallback: anything truthy counts as success.
+        return bool(response)
+
     def switch_compression_strategy(self, compression_strategy):
         if compression_strategy in STRATEGIES:
             self.compression_strategy_name = compression_strategy
@@ -46,7 +65,7 @@ class Segmentation:
             start = time.time()
         data = {"operation": "add_image_prompt", "object_name": object_name, "object_image": object_image}
         response = self.client.send_data(data)
-        if response.get("result") == "SUCCESS":
+        if self._is_success(response):
             self.image_prompt_names.add(object_name)
         if self.benchmark:
             self.call_time["add_image_prompt"] += time.time() - start
@@ -72,7 +91,7 @@ class Segmentation:
             "compression_strategy": self.compression_strategy_name,
         }
         response = self.client.send_data(data)
-        if response.get("result") == "SUCCESS":
+        if self._is_success(response):
             self.first_frame_registered = True
             self.tracking_object_ids = response["data"]["obj_ids"]
             masks = {}
@@ -98,7 +117,7 @@ class Segmentation:
         if self.benchmark:
             start = time.time()
         response = self.client.send_data({"operation": "get_next", "frame": self.compression_strategy.encode(frame)})
-        if response.get("result") == "SUCCESS":
+        if self._is_success(response):
             masks = {}
             for i, obj_id in enumerate(self.tracking_object_ids):
                 mask = self.compression_strategy.decode(response["data"]["masks"][i])

{neuromeka_vfm-0.1.2.dist-info → neuromeka_vfm-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuromeka_vfm
-Version: 0.1.2
+Version: 0.1.3
 Summary: Client utilities for Neuromeka VFM FoundationPose RPC (upload meshes, call server)
 Author: Neuromeka
 License: MIT License

{neuromeka_vfm-0.1.2.dist-info → neuromeka_vfm-0.1.3.dist-info}/RECORD

@@ -2,13 +2,13 @@ neuromeka_vfm/__init__.py,sha256=h5ODdWFgN7a9TBzF6Qfdyx5VxUr2hG0pFTwq57jEvDo,422
 neuromeka_vfm/compression.py,sha256=d2xOz4XBJZ60pPSXwQ5LPYwhpsaNORvNoY_0CUiAvt0,5191
 neuromeka_vfm/pickle_client.py,sha256=Iw2fpxdnKB20oEUgsd0rJlvzOd5JhetphpKkF9qQcX0,591
 neuromeka_vfm/pose_estimation.py,sha256=3MUVhL0nMcpHApZDAzutS7fINPHcb-tu_WoXvNGU33E,2625
-neuromeka_vfm/segmentation.py,sha256=PKOJiNoZz8tlwSX-eAgvqA2YWKkcCkKcoQfaiIEqCdU,5630
+neuromeka_vfm/segmentation.py,sha256=wae0_m225DUMD8Nm2A7iQm49QWeUas17B8PaoGGFt5w,6311
 neuromeka_vfm/upload_mesh.py,sha256=aW5G9aE5OeiDN5pEVKDzMeV538U-I2iRYZvVZTfGsr4,2728
 neuromeka_vfm/examples/__init__.py,sha256=dEhb0FqhpEGNmg0pMunmrTlViIcxvd95fYEjZ49IOTQ,37
 neuromeka_vfm/examples/pose_demo.py,sha256=zq1Z0_kxQc4CB-ltfwm_oMoC7JLoN5GyeE3C6jKGQKw,13658
-neuromeka_vfm-0.1.2.dist-info/licenses/LICENSE,sha256=40cBWxFahhu0p_EB0GhU8oVIifVNmH1o2fZtx0bIif8,1076
-neuromeka_vfm-0.1.2.dist-info/METADATA,sha256=MxiWRP0T5iZzHMsHqyIXSX9JZQF0Kv2dGBhzyOT6EMc,7372
-neuromeka_vfm-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuromeka_vfm-0.1.2.dist-info/entry_points.txt,sha256=Wl4XqiUt_GLQ08oTJtsYjLW0iYxZ52ysVd1-cN0kYP4,72
-neuromeka_vfm-0.1.2.dist-info/top_level.txt,sha256=uAH_yXikUvxXTSEnUC0M8Zl5ggxbnkYtXlmTfEG8MUk,14
-neuromeka_vfm-0.1.2.dist-info/RECORD,,
+neuromeka_vfm-0.1.3.dist-info/licenses/LICENSE,sha256=40cBWxFahhu0p_EB0GhU8oVIifVNmH1o2fZtx0bIif8,1076
+neuromeka_vfm-0.1.3.dist-info/METADATA,sha256=iHvDBzm7K5TqsP13HDSSpJB1o-XL0TJ4Yl77_ueCQac,7372
+neuromeka_vfm-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuromeka_vfm-0.1.3.dist-info/entry_points.txt,sha256=Wl4XqiUt_GLQ08oTJtsYjLW0iYxZ52ysVd1-cN0kYP4,72
+neuromeka_vfm-0.1.3.dist-info/top_level.txt,sha256=uAH_yXikUvxXTSEnUC0M8Zl5ggxbnkYtXlmTfEG8MUk,14
+neuromeka_vfm-0.1.3.dist-info/RECORD,,