teradataml 20.0.0.5__py3-none-any.whl → 20.0.0.7__py3-none-any.whl

teradataml/sdk/spinner.py

@@ -0,0 +1,107 @@
+# ################################################################################################
+# 
+# Copyright 2025 Teradata. All rights reserved.
+# TERADATA CONFIDENTIAL AND TRADE SECRET
+# 
+# Primary Owner: Adithya Avvaru (adithya.avvaru@teradata.com)
+# Secondary Owner: Pankaj Purandare (pankajvinod.purandare@teradata.com)
+# 
+# Version: 1.0
+# ModelOps SDK Version: 1.0
+#
+# This file contains the code for spinner which spins during the SDK execution time.
+# NOTE: This is taken from AoA SDK and will be updated/removed in future.
+#
+# ################################################################################################
+
+import os
+import sys
+import threading
+
+os.system("")  # enables ansi escape characters in terminal
+
+CODE = {
+    "CYAN": "\033[36m",
+    "END": "\033[0m",
+    "RM_LINE": "\033[2K\r",
+    "HIDE_CURSOR": "\033[?25l",
+    "SHOW_CURSOR": "\033[?25h",
+    "CURSOR_NEXT_LINE": "\033[1E",
+    "CURSOR_INIT": "\033[0G",
+}
+
+
+class ProgressBase(threading.Thread):
+    inplace = None
+    stopFlag = None
+
+    def __init__(self):
+        self.rlock = threading.RLock()
+        self.cv = threading.Condition()
+        threading.Thread.__init__(self)
+        self.setDaemon(True)
+
+    def __call__(self):
+        self.start()
+
+    def back_step(self):
+        if self.inplace:
+            sys_print(CODE["CURSOR_NEXT_LINE"])
+
+    def remove_line(self):
+        if self.inplace:
+            sys_print(CODE["RM_LINE"])
+
+    def start(self):
+        self.stopFlag = 0
+        threading.Thread.start(self)
+
+    def stop(self):
+        self.stopFlag = 1
+        sys_print(CODE["SHOW_CURSOR"])
+        self.cv.acquire()
+        self.cv.notify()
+        self.cv.release()
+        self.rlock.acquire()
+
+
+class Spinner(ProgressBase):
+
+    def __init__(self, msg="", speed=0.1):
+        self.__seq = ["⣾", "⣷", "⣯", "⣟", "⡿", "⢿", "⣻", "⣽"]
+        self.__speed = speed
+        self.__msg = msg
+        self.inplace = 1
+        ProgressBase.__init__(self)
+
+    def run(self):
+        self.rlock.acquire()
+        self.cv.acquire()
+        sys_print(CODE["HIDE_CURSOR"])
+        while 1:
+            for char in self.__seq:
+                self.cv.wait(self.__speed)
+                if self.stopFlag:
+                    self.back_step()
+                    try:
+                        return
+                    finally:
+                        self.rlock.release()
+                if self.inplace:
+                    sys_print(
+                        f"{CODE['CYAN']}{char}{CODE['END']} {self.__msg}{CODE['CURSOR_INIT']}"
+                    )
+
+
+def sys_print(msg):
+    sys.stdout.write(msg)
+    sys.stdout.flush()
+
+
+def spin_it(function, msg, speed=0.25, *args, **kwargs):
+    indicator = Spinner(msg, speed)
+    indicator.start()
+    result = function(*args, **kwargs)
+    indicator.stop()
+    indicator.remove_line()
+    return result

teradataml/series/series.py

@@ -78,6 +78,7 @@ class Series:
         # Optional arguments
         self._df_orderby = kw.pop('order_by', None)
         self._df_index_label = kw.pop('index_label', None)
+        self._datalake = kw.pop('datalake', None)
 
         # Derived attributes
         # Note that this is considering we have only one column,
@@ -92,7 +93,8 @@ class Series:
             self._aed_utils = AedUtils()
             self._metaexpr = UtilFuncs._get_metaexpr_using_columns(nodeid,
                                                                    zip([self._name],
-                                                                       [self._col.type]))
+                                                                       [self._col.type]),
+                                                                   datalake=self._datalake)
 
         except TeradataMlException:
             raise
@@ -101,7 +103,7 @@ class Series:
                                       MessageCodes.SERIES_CREATE_FAIL) from err
 
     @classmethod
-    def _from_node(cls, axis, nodeid, col, df_orderby = None, df_index_label = None):
+    def _from_node(cls, axis, nodeid, col, df_orderby = None, df_index_label = None, **kw):
         """
         DESCRIPTION:
             Private class method for creating a teradataml Series from a nodeid, and parent metadata.
@@ -149,7 +151,8 @@ class Series:
                      nodeid,
                      col,
                      order_by = df_orderby,
-                     index_label = df_index_label)
+                     index_label = df_index_label,
+                     datalake=kw.pop("datalake", None))
 
         return series
 
@@ -195,8 +198,9 @@ class Series:
         series = cls(axis,
                      df._nodeid,
                      df._metaexpr.c[0],
-                     order_by = df._orderby,
-                     index_label = df._index_label)
+                     order_by=df._orderby,
+                     index_label=df._index_label,
+                     datalake=df._metaexpr.datalake)
 
         return series
 
@@ -244,7 +248,6 @@ class Series:
         """
         # TODO : Change docstring based on implementation of RowBased Series object
         query = repr(self._metaexpr) + ' FROM ' + self._get_table_name()
-
         if self._df_orderby is not None:
             query += ' ORDER BY ' + self._df_orderby
 
@@ -449,7 +452,9 @@ class Series:
                                                      assign_expression,
                                                      AEDConstants.AED_ASSIGN_DROP_EXISITING_COLUMNS.value)
 
-            return Series._from_node(self._axis, new_nodeid, new_meta.c[0], self._df_orderby, self._df_index_label)
+            return Series._from_node(self._axis, new_nodeid, new_meta.c[0],
+                                     self._df_orderby, self._df_index_label,
+                                     datalake=self._metaexpr.datalake)
 
         except TeradataMlException:
             raise