1 files changed, 48 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_utils/_azureml_polling.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_utils/_azureml_polling.py
new file mode 100644
index 00000000..fc148426
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_utils/_azureml_polling.py
@@ -0,0 +1,48 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+
+import logging
+from concurrent.futures import Future
+from time import time
+from typing import Optional, Union
+
+from azure.ai.ml.constants._common import LROConfigurations
+from azure.core.polling import LROPoller
+from azure.mgmt.core.polling.arm_polling import ARMPolling
+
+module_logger = logging.getLogger(__name__)
+
+
+class AzureMLPolling(ARMPolling):
+    """A polling class for azure machine learning."""
+
+    def update_status(self):
+        """Update the current status of the LRO."""
+        super().update_status()
+        print(".", end="", flush=True)
+
+
+def polling_wait(
+    poller: Union[LROPoller, Future],
+    message: Optional[str] = None,
+    start_time: Optional[float] = None,
+) -> None:
+    """Print out status while polling and time of operation once completed.
+
+    :param poller: An poller which will return status update via function done().
+    :type poller: Union[LROPoller, concurrent.futures.Future]
+    :param message: Message to print out before starting operation write-out.
+    :type message: Optional[str]
+    :param start_time: Start time of operation.
+    :type start_time: Optional[float]
+    """
+    module_logger.info("%s", message)
+    poller.result(timeout=LROConfigurations.POLLING_TIMEOUT)
+    module_logger.info("Done.")
+
+    if start_time:
+        end_time = time()
+        duration = divmod(int(round(end_time - start_time)), 60)
+        module_logger.info("(%sm %ss)\n", duration[0], duration[1])