Fix update_info problem

Superb-AI-Suite · Nov 10, 2023 · 2aae8ce · 2aae8ce
1 parent 0467c4a
commit 2aae8ce
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/spb/labels/manager.py b/spb/labels/manager.py
@@ -340,15 +340,15 @@ def update_info(
                 result=result
             )
             response = self.session.execute(query, values)
+            data = self.session.get_data_from_mutation(response, QUERY_ID)
+            updated_label = Label(**data)
             self.set_info_with_url(
                 label_info=label_info,
-                label=label
+                label=updated_label
             )
+            updated_label = self.session.get_label_info_from_url(updated_label)
         except Exception as e:
             raise e
-        updated_label = self.session.build_label_from_response(
-            QUERY_ID, response
-        )
         return updated_label
 
     def update_tags(self, label: Label, tags: List[Union[str, Tags]]):

diff --git a/spb/sdk_config.py b/spb/sdk_config.py
@@ -1,6 +1,6 @@
 SDK_NAME='spb-cli'
 SDK_AUTHOR='Superb AI Dev Team'
-SDK_VERSION='0.18.0'
+SDK_VERSION='0.18.1'
 SDK_AUTHOR_EMAIL='support@superb-ai.com'
 SDK_DESCRIPTION='Suite Standard Library'
 SDK_LICENSE='MIT'