🐛 Slice into array when copying

mlexchange · Mar 9, 2024 · a27b6cd · a27b6cd
1 parent 8b53e25
commit a27b6cd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/examples/copy_mask_as_result.py b/examples/copy_mask_as_result.py
@@ -14,11 +14,15 @@
 
 def copy_mask_as_result(mask_uri, job_id, quick_inference=True):
     mask_client = from_uri(mask_uri, api_key=SEG_TILED_API_KEY)
-    mask = mask_client["mask"]
+    mask = mask_client["mask"][:]
     mask_metadata = mask_client.metadata
 
+    print(
+        f"Copying mask with unique values: {np.unique(mask)} and shape: {mask.shape}."
+    )
+
     result_client = from_uri(SEG_TILED_URI, api_key=SEG_TILED_API_KEY)
-    print(result_client)
+
     result_container = result_client.create_container(key=job_id)
     result_metadata = {
         "data_uri": mask_metadata["data_uri"],