orca-sdk 0.0.88__tar.gz → 0.0.90__tar.gz

{orca_sdk-0.0.88 → orca_sdk-0.0.90}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: orca_sdk
-Version: 0.0.88
+Version: 0.0.90
 Summary: SDK for interacting with Orca Services
 License: Apache-2.0
 Author: Orca DB Inc.

{orca_sdk-0.0.88 → orca_sdk-0.0.90}/orca_sdk/_utils/analysis_ui.py

@@ -148,6 +148,10 @@ def display_suggested_memory_relabels(memoryset: LabeledMemoryset):
                         )
                 for i, memory_relabel in enumerate(current_memory_relabel_map.values()):
                     mem = memory_relabel["full_memory"]
+                    predicted_label = mem.metrics["neighbor_predicted_label"]
+                    predicted_label_name = label_names[predicted_label]
+                    predicted_label_confidence = mem.metrics.get("neighbor_predicted_label_confidence", 0)
+
                     with gr.Row(equal_height=True, variant="panel", elem_classes="white" if i % 2 == 0 else None):
                         with gr.Column(scale=9):
                             assert isinstance(mem.value, str)
@@ -158,12 +162,12 @@ def display_suggested_memory_relabels(memoryset: LabeledMemoryset):
                             dropdown = gr.Dropdown(
                                 choices=[f"{label_name} ({i})" for i, label_name in enumerate(label_names)],
                                 label="SuggestedLabel",
-                                value=f"{label_names[mem.metrics.neighbor_predicted_label]} ({mem.metrics.neighbor_predicted_label})",
+                                value=f"{predicted_label_name} ({predicted_label})",
                                 interactive=True,
                                 container=False,
                             )
                             confidence = gr.HTML(
-                                f"<p style='font-size: 10px; color: #888;'>Confidence: {mem.metrics.neighbor_predicted_label_confidence:.2f}</p>",
+                                f"<p style='font-size: 10px; color: #888;'>Confidence: {predicted_label_confidence:.2f}</p>",
                                 elem_classes="no-padding",
                             )
                             dropdown.change(

{orca_sdk-0.0.88 → orca_sdk-0.0.90}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "orca_sdk"
-version = "0.0.88" # Will be set by CI before building the wheel from the git tag. Do not set manually.
+version = "0.0.90" # Will be set by CI before building the wheel from the git tag. Do not set manually.
 description = "SDK for interacting with Orca Services"
 authors = ["Orca DB Inc. <dev-rel@orcadb.ai>"]
 license = "Apache-2.0"