upgini 1.1.305__py3-none-any.whl → 1.1.306__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.1.305"
+__version__ = "1.1.306"

upgini/data_source/data_source_publisher.py

@@ -418,3 +418,19 @@ class DataSourcePublisher:
         response = self._rest_client.union_search_tasks(request, "trace_id")
         print(response)
         return response
+
+    def reannounce_all_ads(self):
+        trace_id = str(uuid.uuid4())
+        with MDC(trace_id=trace_id):
+            try:
+                task_id = self._rest_client.reannounce_all_ads(trace_id)
+                with Spinner():
+                    status_response = self._rest_client.poll_ads_management_task_status(task_id, trace_id)
+                    while status_response["status"] not in self.FINAL_STATUSES:
+                        time.sleep(5)
+                        status_response = self._rest_client.poll_ads_management_task_status(task_id, trace_id)
+
+                if status_response["status"] != "COMPLETED":
+                    raise Exception("Failed to reannounce all ADS: " + status_response["errorMessage"])
+            except Exception:
+                self.logger.exception("Failed to reannounce all ADS-es")

upgini/http.py

@@ -285,6 +285,7 @@ class _RestClient:
     DEACTIVATE_ADS_URI = "private/api/v2/ads/deactivate"
     TOGGLE_ADS_URI_FMT = "private/api/v2/ads/{0}/toggle"
     DELETE_ADS_URI_FMT = "private/api/v2/ads/{0}"
+    REANNOUNCE_ALL_ADS_URI_FMT = "private/api/v2/ads/reannounce-all-ads"
     POLL_ADS_MANAGEMENT_STATUS_URI_FMT = "private/api/v2/ads/management-task/{0}"
     GET_ADS_DESCRIPTION_URI_FMT = "private/api/v2/ads/{0}"
     GET_ALL_ADS_DESCRIPTIONS_URI = "private/api/v2/ads/descriptions"
@@ -726,6 +727,11 @@ class _RestClient:
         api_path = self.DEACTIVATE_ADS_URI
         self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id, request, result_format=None))
 
+    def reannounce_all_ads(self, trace_id: str) -> str:
+        api_path = self.REANNOUNCE_ALL_ADS_URI_FMT
+        response = self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id))
+        return response["adsManagementTaskId"]
+
     def toggle_ads(self, ads_definition_id: str, trace_id: str):
         api_path = self.TOGGLE_ADS_URI_FMT.format(ads_definition_id)
         return self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id))

{upgini-1.1.305.dist-info → upgini-1.1.306.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.305
+Version: 1.1.306
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

{upgini-1.1.305.dist-info → upgini-1.1.306.dist-info}/RECORD

@@ -1,10 +1,10 @@
-upgini/__about__.py,sha256=QMF-ZjUcnh9mDz_6F2wxDw2kRy9dur-Ngw1eE7MTiR4,24
+upgini/__about__.py,sha256=okL1wSisFFnkKhJ4nbM09p9rNtG-hKV5SxZSxyhmBQk,24
 upgini/__init__.py,sha256=ObEtjFkIssl83qeKNMLpIQygfwK8TzztwiI43YTsAP0,353
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=MOzBVsvzlHLxNfPWtMaXC_jIPeW7_gUvbSGeXnsPgNI,46158
 upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
 upgini/features_enricher.py,sha256=8grYkWdix0NogYAY7YjUVe2Tqfra6NAxM78N9MlLF30,183572
-upgini/http.py,sha256=bp6jWl422Icy3AhHMdCcJv5NjExE45gSMmzMTPJjPuk,42600
+upgini/http.py,sha256=a4Epc9YLIJBuYk4t8E_2-QDLBtJFqKO35jn2SnYQZCg,42920
 upgini/lazy_import.py,sha256=EwoM0msNGbSmWBhGbrLDny1DSnOlvTxCjmMKPxYlDms,610
 upgini/metadata.py,sha256=E5WWZ_MkjGyYNQh_LnwMIBHyqPx1fxk-qhEfQIJnzq8,10209
 upgini/metrics.py,sha256=Tu5cN8RlhOSSMWUTXRSkdl8SWBqR1N_2eJpBum9pZxc,30926
@@ -23,7 +23,7 @@ upgini/autofe/operand.py,sha256=MKEsl3zxpWzRDpTkE0sNJxTu62U20sWOvEKhPjUWS6s,2915
 upgini/autofe/unary.py,sha256=ZWjLd-CUkNt_PpM8YuWLLipW1v_RdBlsl4JxXIVo9aM,3652
 upgini/autofe/vector.py,sha256=dLxfAstJs-gw_OQ1xxoxcM6pVzORlV0HVzdzt7cLXVQ,606
 upgini/data_source/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-upgini/data_source/data_source_publisher.py,sha256=aCVpZ3RGlYToRYYNHjC2BxIxnnKUyrrF29MncCXhVlE,19716
+upgini/data_source/data_source_publisher.py,sha256=y0tYFp7N3bSI7BwQ5SRF8r0bRaI3z6Zc1fsZezVg7hE,20552
 upgini/mdc/__init__.py,sha256=aM08nIWFc2gWdWUa3_IuEnNND0cQPkBGnYpRMnfFN8k,1019
 upgini/mdc/context.py,sha256=3u1B-jXt7tXEvNcV3qmR9SDCseudnY7KYsLclBdwVLk,1405
 upgini/normalizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -57,7 +57,7 @@ upgini/utils/sklearn_ext.py,sha256=13jQS_k7v0aUtudXV6nGUEWjttPQzAW9AFYL5wgEz9k,4
 upgini/utils/target_utils.py,sha256=Y96_PJ5cC-WsEbeqg20v9uqywDQobLoTb-xoP7S3o4E,7807
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.1.305.dist-info/METADATA,sha256=xeEcJbbZ8LrhTyzLvyGR9I5FsVC2gP1Rn65cPo3gJrg,48153
-upgini-1.1.305.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.1.305.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.305.dist-info/RECORD,,
+upgini-1.1.306.dist-info/METADATA,sha256=GUAb3dq_ebpqOqGoH4gEydc7T1Z91cLIur9nMvmrUEg,48153
+upgini-1.1.306.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.1.306.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.306.dist-info/RECORD,,

{upgini-1.1.305.dist-info → upgini-1.1.306.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.24.2
+Generator: hatchling 1.25.0
 Root-Is-Purelib: true
 Tag: py3-none-any