dora-sam2 0.3.10__py3-none-any.whl → 0.3.11__py3-none-any.whl

dora_sam2/__init__.py

@@ -1,3 +1,5 @@
+"""TODO: Add docstring."""
+
 import os
 
 # Define the path to the README file relative to the package directory

dora_sam2/__main__.py

@@ -1,3 +1,5 @@
+"""TODO: Add docstring."""
+
 from .main import main
 
 if __name__ == "__main__":

dora_sam2/main.py

@@ -1,3 +1,5 @@
+"""TODO: Add docstring."""
+
 import cv2
 import numpy as np
 import pyarrow as pa
@@ -10,6 +12,7 @@ predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
 
 
 def main():
+    """TODO: Add docstring."""
     pa.array([])  # initialize pyarrow array
     node = Node()
     frames = {}
@@ -119,8 +122,8 @@ def main():
                                             {
                                                 "masks": masks.ravel(),
                                                 "labels": event["value"]["labels"],
-                                            }
-                                        ]
+                                            },
+                                        ],
                                     ),
                                     metadata={
                                         "image_id": image_id,
@@ -129,7 +132,7 @@ def main():
                                     },
                                 )
 
-            elif "boxes2d" in event_id:
+            if "boxes2d" in event_id:
 
                 if isinstance(event["value"], pa.StructArray):
                     boxes2d = event["value"][0].get("bbox").values.to_numpy()
@@ -159,7 +162,7 @@ def main():
                 ):
                     predictor.set_image(frames[image_id])
                     masks, _scores, last_pred = predictor.predict(
-                        box=boxes2d, point_labels=labels, multimask_output=False
+                        box=boxes2d, point_labels=labels, multimask_output=False,
                     )
 
                     if len(masks.shape) == 4:
@@ -190,8 +193,8 @@ def main():
                                         {
                                             "masks": masks.ravel(),
                                             "labels": event["value"]["labels"],
-                                        }
-                                    ]
+                                        },
+                                    ],
                                 ),
                                 metadata={
                                     "image_id": image_id,

{dora_sam2-0.3.10.dist-info → dora_sam2-0.3.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: dora-sam2
-Version: 0.3.10
+Version: 0.3.11
 Summary: dora-sam2
 Author-email: Your Name <email@email.com>
 License: MIT

dora_sam2-0.3.11.dist-info/RECORD

@@ -0,0 +1,8 @@
+dora_sam2/__init__.py,sha256=tF7WHhHiDweUUzyHsbmFe_ktphE08aA5j33E4ja1udA,381
+dora_sam2/__main__.py,sha256=NlAb6Jbmmn82K8Ahdi12sliZYdzyY7QaoCCHRuoR_Hg,90
+dora_sam2/main.py,sha256=aQj1zZTtpzebwaayyPqD4B82iFscLKuZ5v7COs3dTNE,8487
+dora_sam2-0.3.11.dist-info/METADATA,sha256=jTkg2BE_lLdUTroTRkZ6hdsT2xwwtF9Z2D_SYtgG8_Y,820
+dora_sam2-0.3.11.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+dora_sam2-0.3.11.dist-info/entry_points.txt,sha256=eObMaDQauVA_sv4J6fsNjn8V-8syGJK7mO-LrsBu1aA,50
+dora_sam2-0.3.11.dist-info/top_level.txt,sha256=IgKcOITGe2Nlyc79J6dwh3dcp3Wsf-IipIy-1h9GcPE,10
+dora_sam2-0.3.11.dist-info/RECORD,,

{dora_sam2-0.3.10.dist-info → dora_sam2-0.3.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.2)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

dora_sam2-0.3.10.dist-info/RECORD

@@ -1,8 +0,0 @@
-dora_sam2/__init__.py,sha256=HuSK3dnyI9Pb5QAuaKFwQQ3J5SIZnLcKHPJO0norGzc,353
-dora_sam2/__main__.py,sha256=Vdhw8YA1K3wPMlbJQYL5WqvRzAKVeZ16mZQFO9VRmCo,62
-dora_sam2/main.py,sha256=zwMJwenWZdmpMfriS8C903g7QdO-qOJrn2lP4PCcPyo,8425
-dora_sam2-0.3.10.dist-info/METADATA,sha256=bC15e7QBHnl8nyT-E5tQLoPdI9LAviaLNsUD7ZglJB0,820
-dora_sam2-0.3.10.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-dora_sam2-0.3.10.dist-info/entry_points.txt,sha256=eObMaDQauVA_sv4J6fsNjn8V-8syGJK7mO-LrsBu1aA,50
-dora_sam2-0.3.10.dist-info/top_level.txt,sha256=IgKcOITGe2Nlyc79J6dwh3dcp3Wsf-IipIy-1h9GcPE,10
-dora_sam2-0.3.10.dist-info/RECORD,,