orca-sdk 0.0.88__py3-none-any.whl → 0.0.90__py3-none-any.whl

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.dist-info → orca_sdk-0.0.90.dist-info}/METADATA

@@ -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.dist-info → orca_sdk-0.0.90.dist-info}/RECORD

@@ -185,7 +185,7 @@ orca_sdk/_generated_api_client/models/update_prediction_request.py,sha256=HMPq_K
 orca_sdk/_generated_api_client/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
 orca_sdk/_generated_api_client/types.py,sha256=j7-uA7wWwN1cq0d7ULccN4vDm-1IzgnrxSyVktxvABM,1399
 orca_sdk/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-orca_sdk/_utils/analysis_ui.py,sha256=SzbeeQbi6dbWkE0Awo-8qStxLolU6LaZzIGdPc6Tars,9075
+orca_sdk/_utils/analysis_ui.py,sha256=DE_fO1XE1KGEqhZRgmXrwVKi-ZRVSkgD80FqrkSsV_s,9262
 orca_sdk/_utils/analysis_ui_style.css,sha256=OhbSsOaz1d-v8CIxKDsFO0hVVk2Afx6WOAz2KzA8EwQ,874
 orca_sdk/_utils/auth.py,sha256=9nLgcuw5B9BCkThN8Qhyq0TeJAyaRSK4bLbOsZCDIqg,2793
 orca_sdk/_utils/auth_test.py,sha256=ygVWv1Ex53LaxIP7p2hzPHl8l9qYyBD5IGmEFJMps6s,1056
@@ -210,6 +210,6 @@ orca_sdk/memoryset.py,sha256=NX31cPfdG_a7A9aw0Op_1T1FGwDvpV33XKgDEwGHOPI,55479
 orca_sdk/memoryset_test.py,sha256=7X_b6HNq28kZdyvMxYUtiajrWoXm0Z0B4qoUWpx3cFU,14296
 orca_sdk/telemetry.py,sha256=Nz8fSDR6EUeNfGpe50I98REyNnqR9vJulBMfd9skIE0,15921
 orca_sdk/telemetry_test.py,sha256=p76MHP2LEMHQ4ssCbd3ImPyt_CoULiwi13FoOvkAV1w,4785
-orca_sdk-0.0.88.dist-info/METADATA,sha256=NN8TLpamM1qWTFl-fFvlpJ0tL-tgI8CjEsH9BfY7FKE,3138
-orca_sdk-0.0.88.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-orca_sdk-0.0.88.dist-info/RECORD,,
+orca_sdk-0.0.90.dist-info/METADATA,sha256=jTJ2GTd2n_hnWybPOZIaWCmmXGQgUVgNc0v4YuWDMQ0,3138
+orca_sdk-0.0.90.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+orca_sdk-0.0.90.dist-info/RECORD,,