habana-pyhlml 1.21.4.3__py3-none-any.whl → 1.22.1.6__py3-none-any.whl

pyhlml/main.py

@@ -11,29 +11,18 @@ from pyhlml.hlml_error import HLMLError, ErrorsAsClass
 _hlmlOBJ = None
 
 def _wrapperInit():
-    """ init module level object and error classes
-        Parameters: None
-        Returns: None
-    """
+    """Initialize module-level HLML object and error classes."""
     ErrorsAsClass()
     setattr(modules[__name__], "_hlmlOBJ", LibHLML())
 
 def check_return(ret):
-    """ Checks ret for any error.
-        Parameters:
-            ret - Symbol for checking if return was valid.
-        Returns:
-            ret - Symbol for current error status.
-    """
+    """Raise HLMLError if ret is not HLML_SUCCESS, otherwise return ret."""
     if (ret != hlml_t.HLML_RETURN.HLML_SUCCESS ):
         raise HLMLError(ret)
     return ret
 
 def hlmlInit() -> None:
-    """ Must be called before any other api can be used
-        Parameters: None
-        Returns: None
-    """
+    """Initialize HLML. Must be called before any other API can be used."""
     _wrapperInit()
     global _hlmlOBJ
 
@@ -45,10 +34,9 @@ def hlmlInit() -> None:
     return None
 
 def hlmlInitWithFlags(flags=0) -> None:
-    """ Allows the user to call the init function with a flag.
-        Parameters:
-            flags (int) [ default=0 ] - only the default is supported
-        Returns: None
+    """Initialize HLML with flags. Only the default flag (0) is supported.
+    Args:
+        flags (int): Initialization flags (default 0).
     """
     _wrapperInit()
     global _hlmlOBJ
@@ -61,10 +49,7 @@ def hlmlInitWithFlags(flags=0) -> None:
     return None
 
 def hlmlShutdown() -> None:
-    """ Shutdown should be called last.
-        Parameters: None
-        Returns: None
-    """
+    """Shutdown HLML and release resources. Should be called last."""
     global _hlmlOBJ
 
     fn = _hlmlOBJ.get_func_ptr("hlml_shutdown")
@@ -76,11 +61,9 @@ def hlmlShutdown() -> None:
     return None
 
 def hlmlDeviceGetCount() -> int:
-    """ Returns the number of Habana devices in the system.
-        Parameters: None
-        Returns:
-            count (int) - Number of habana devices.
-                          Ex. An HLS1-H would return 4
+    """Get the number of Habana devices in the system.
+    Returns:
+        int: Number of Habana devices.
     """
     global _hlmlOBJ
     count = ctypes.c_uint()
@@ -92,11 +75,13 @@ def hlmlDeviceGetCount() -> int:
     return count.value
 
 def hlmlDeviceGetHandleByPCIBusID(pci_addr: str) -> hlml_t.HLML_DEVICE.TYPE:
-    """ Acquire the handle for a device, based on PCI Address
-        Parameters:
-            pci_addr (str) - The PCI Address of a habana device.
-        Returns:
-            device (HLML_DEVICE.TYPE) - The handle for the habana device.
+    """Get device handle by PCI address.
+        If there is a single domain, the 'domain' part of the PCI address
+        is not required.
+    Args:
+        pci_addr (str): PCI address of the device.
+    Returns:
+        HLML_DEVICE.TYPE: Device handle.
     """
     global _hlmlOBJ
 
@@ -108,11 +93,11 @@ def hlmlDeviceGetHandleByPCIBusID(pci_addr: str) -> hlml_t.HLML_DEVICE.TYPE:
     return device
 
 def hlmlDeviceGetHandleByIndex(index: int) -> hlml_t.HLML_DEVICE.TYPE:
-    """ Acquire device handle by index
-        Parameters:
-            index (int) - The index of a habana device.
-        Returns:
-            device (HLML_DEVICE.TYPE) - The handle for the habana device.
+    """Get device handle by index.
+    Args:
+        index (int): Device index.
+    Returns:
+        HLML_DEVICE.TYPE: Device handle.
     """
     global _hlmlOBJ
     device = hlml_t.HLML_DEVICE.TYPE()
@@ -124,11 +109,11 @@ def hlmlDeviceGetHandleByIndex(index: int) -> hlml_t.HLML_DEVICE.TYPE:
     return device
 
 def hlmlDeviceGetHandleByUUID(uuid: str) -> hlml_t.HLML_DEVICE.TYPE:
-    """ Acquire device handle by UUID
-        Parameters:
-            uuid (str) - A Universal Unique ID for a habana device to access.
-        Returns:
-            device (HLML_DEVICE.TYPE) - The handle for the habana device.
+    """Get device handle by UUID (Universal Unique ID).
+    Args:
+        uuid (str): Device UUID.
+    Returns:
+        HLML_DEVICE.TYPE: Device handle.
     """
     global _hlmlOBJ
     device = hlml_t.HLML_DEVICE.TYPE()
@@ -139,12 +124,12 @@ def hlmlDeviceGetHandleByUUID(uuid: str) -> hlml_t.HLML_DEVICE.TYPE:
     check_return(ret)
     return device
 
-def hlmlDeviceGetName(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Acquire name of device from handle
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            name (str) - The name of the habana device.
+def hlmlDeviceGetName(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get device name from handle.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Device name.
     """
     global _hlmlOBJ
     name = ctypes.create_string_buffer(hlml_t.HLML_DEFINE.HL_FIELD_MAX_SIZE)
@@ -159,11 +144,11 @@ def hlmlDeviceGetName(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     return name.value
 
 def hlmlDeviceGetPCIInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_pci_info:
-    """ Get the PCI attributes of input device
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            pci_info (c_hlml_pci_info) - The PCI attributes of the given device.
+    """Get PCI attributes of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        c_hlml_pci_info: PCI attributes.
     """
     global _hlmlOBJ
     pci_info = hlml_t.c_hlml_pci_info()
@@ -178,12 +163,12 @@ def hlmlDeviceGetPCIInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_pci_i
     return pci_info
 
 def hlmlDeviceGetClockInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) -> int:
-    """ Retrives the current speed of the selected clock (MHz)
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            clock_type ( 0-SOC ( GAUDI ) / 1-IC ( GOYA ) / 2-MME ( GOYA ) / 3-TPC ( GOYA ) )
-        Returns:
-            speed (int) - The clock speed of the selected clock in MHz.
+    """Get current speed of the selected clock (MHz).
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        clock_type (int): Clock type.
+    Returns:
+        int: Clock speed in MHz.
     """
     global _hlmlOBJ
     speed = ctypes.c_uint()
@@ -195,12 +180,12 @@ def hlmlDeviceGetClockInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) -> in
     return speed.value
 
 def hlmlDeviceGetMaxClockInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) -> int:
-    """ Retrives the maximum speed of the selected clock (MHz)
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            clock_type ( 0-SOC ( GAUDI ) / 1-IC ( GOYA ) / 2-MME ( GOYA ) / 3-TPC ( GOYA ) )
-        Returns:
-            speed (int) - The MAXIMUM clock speed of the selected clock in MHz.
+    """Get maximum speed of the selected clock (MHz).
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        clock_type (int): Clock type.
+    Returns:
+        int: Maximum clock speed in MHz.
     """
     global _hlmlOBJ
     speed = ctypes.c_uint()
@@ -212,12 +197,12 @@ def hlmlDeviceGetMaxClockInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) ->
     return speed.value
 
 def hlmlDeviceGetClockLimitInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) -> int:
-    """ Retrives the frequency limit of the selected clock (MHz)
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            clock_type ( 0-SOC ( GAUDI ) / 1-IC ( GOYA ) / 2-MME ( GOYA ) / 3-TPC ( GOYA ) )
-        Returns:
-            speed (int) - The frequency LIMIT of the selected clock in MHz.
+    """Get frequency limit of the selected clock (MHz).
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        clock_type (int): Clock type.
+    Returns:
+        int: Frequency limit in MHz.
     """
     global _hlmlOBJ
     speed = ctypes.c_uint()
@@ -229,11 +214,11 @@ def hlmlDeviceGetClockLimitInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 )
     return speed.value
 
 def hlmlDeviceGetUtilizationRates(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_utilization:
-    """ Returns utilization over the past second as a percentage
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            hlml_util (int) - The utilization rate over the last second as a percentage.
+    """Get device utilization rates over the past second.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        c_hlml_utilization: Utilization rates.
     """
     global _hlmlOBJ
     hlml_util = hlml_t.c_hlml_utilization()
@@ -245,11 +230,11 @@ def hlmlDeviceGetUtilizationRates(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_h
     return hlml_util
 
 def hlmlDeviceGetProcessUtilization(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_process_utilization:
-    """ Returns process utilization over the past second as a percentage
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            hlml_util (int) - The utilization rate over the last second as a percentage.
+    """Get process utilization rates (as a percentage) over the past second.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        c_hlml_process_utilization: Process utilization rates.
     """
     global _hlmlOBJ
     hlml_util = hlml_t.c_hlml_process_utilization()
@@ -261,11 +246,11 @@ def hlmlDeviceGetProcessUtilization(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c
     return hlml_util
 
 def hlmlDeviceGetMemoryInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_memory:
-    """ Returns the total, used, and free memory in bytes
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            hlml_mem (c_hlml_memory) - The total memory and how much is used/free in bytes.
+    """Get total, used, and free memory in bytes.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        c_hlml_memory: Memory info.
     """
     global _hlmlOBJ
     hlml_mem = hlml_t.c_hlml_memory()
@@ -278,12 +263,12 @@ def hlmlDeviceGetMemoryInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_me
 
 def hlmlDeviceGetTemperature(
         device: hlml_t.HLML_DEVICE.TYPE, sensor_type: hlml_t.HLML_TEMP_SENS.TYPE) -> int:
-    """ Retrives the current temperature (celsius) of the selected sensor_type
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            sensor_types ( 0-AIP / 1-BOARD )
-        Returns:
-            temp (int) - The temperature recorded at the given sensor in celsius.
+    """Get current temperature (Celsius) of the selected sensor.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        sensor_type (HLML_TEMP_SENS.TYPE): Sensor type.
+    Returns:
+        int: Temperature in Celsius.
     """
     global _hlmlOBJ
     temp = ctypes.c_uint()
@@ -295,12 +280,12 @@ def hlmlDeviceGetTemperature(
     return temp.value
 
 def hlmlDeviceGetTemperatureThreshold(device: hlml_t.HLML_DEVICE.TYPE, threshold_type: int) -> int:
-    """ Retrieves the known temperature (celsius) threshold of the requested type
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            threshold_type (0-shutdown / 1-slowdown / 2-memory / 3-gpu ) -  Which threshold temp to check.
-        Returns:
-            temp (int) - The temperature the given threshold is set at in celsius.
+    """Get temperature threshold (Celsius) of the requested type.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        threshold_type (int): Threshold type.
+    Returns:
+        int: Threshold temperature in Celsius.
     """
     global _hlmlOBJ
     temp = ctypes.c_uint()
@@ -311,12 +296,12 @@ def hlmlDeviceGetTemperatureThreshold(device: hlml_t.HLML_DEVICE.TYPE, threshold
     check_return(ret)
     return temp.value
 
-def hlmlDeviceGetPersistenceMode(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLML_ENABLE_STATE:
-    """ Retrieves the persistence mode of the device
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            mode (int) - The persistence mode of the device.
+def hlmlDeviceGetPersistenceMode(device: hlml_t.HLML_DEVICE.TYPE) -> int:
+    """Get persistence mode of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Persistence mode.
     """
     global _hlmlOBJ
     mode = ctypes.c_uint()
@@ -327,12 +312,12 @@ def hlmlDeviceGetPersistenceMode(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLML
     check_return(ret)
     return mode.value
 
-def hlmlDeviceGetPerformanceState(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLML_P_STATES:
-    """ Retrieves the performance state of the device
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            p_state (int) - The performance state of the device.
+def hlmlDeviceGetPerformanceState(device: hlml_t.HLML_DEVICE.TYPE) -> int:
+    """Get performance state of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Performance state.
     """
     global _hlmlOBJ
     p_state = ctypes.c_uint()
@@ -343,19 +328,16 @@ def hlmlDeviceGetPerformanceState(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLM
     check_return(ret)
     return p_state.value
 
-def hlmlDeviceGetSupportedPerformanceStates( device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLML_P_STATES:
-    """ Get all supported Performance States (HLML_P_STATES) for the device.
-        The returned array will be in size HLML_PSTATE_NUM_SUPPORTED and will contain a contiguous list
-        of valid P-States supported by the device.
-        If the number of supported P-States is fewer than the size of the array
-        missing elements would contain HLML_PSTATE_UNKNOWN.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            p_states (hlml_t.HLML_P_STATES array in size HLML_PSTATE_NUM_SUPPORTED) - that contains
-                    all the supported performance state of the device.
-                    If the number of supported P-States is fewer than the size of the array
-                    missing elements would contain HLML_PSTATE_UNKNOWN.
+def hlmlDeviceGetSupportedPerformanceStates(device: hlml_t.HLML_DEVICE.TYPE) -> tuple:
+    """Get all supported performance states for the device.
+        The returned array is of size HLML_PSTATE_NUM_SUPPORTED and contains a contiguous
+        list of performance states that the device supports.
+        If the number of supported performance states is less than HLML_PSTATE_NUM_SUPPORTED,
+        the remaining entries in the array will be set to HLML_PSTATE_UNKNOWN.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        HLML_P_STATES array: Supported performance states.
     """
     global _hlmlOBJ
     p_states = (hlml_t.HLML_P_STATES.TYPE * hlml_t.HLML_PSTATE_NUM_SUPPORTED)()
@@ -367,11 +349,11 @@ def hlmlDeviceGetSupportedPerformanceStates( device: hlml_t.HLML_DEVICE.TYPE) ->
     return p_states
 
 def hlmlDeviceGetPowerUsage(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves power usage for the device in mW
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            power (int) - The given device's power usage in mW.
+    """Get power usage for the device in mW.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Power usage in mW.
     """
     global _hlmlOBJ
     power = ctypes.c_uint()
@@ -383,12 +365,10 @@ def hlmlDeviceGetPowerUsage(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return power.value
 
 def hlmlDeviceSetPowerManagementLimit(device: hlml_t.HLML_DEVICE.TYPE, limit : ctypes.c_uint) -> None:
-    """ Sets power management limit on this device in mW
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            limit (ctypes.c_uint) - The power limit on the device in mW.
-        Returns:
-            ret (HLML_RETURN) - The status of the operation.
+    """Set power management limit on this device in mW.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        limit (ctypes.c_uint): Power limit in mW.
     """
     global _hlmlOBJ
 
@@ -399,11 +379,11 @@ def hlmlDeviceSetPowerManagementLimit(device: hlml_t.HLML_DEVICE.TYPE, limit : c
     return None
 
 def hlmlDeviceGetPowerManagementDefaultLimit(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves default power management limit on this device in mW
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            power (int) - The power limit on the device in mW.
+    """Get default power management limit on this device in mW.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Default power limit in mW.
     """
     global _hlmlOBJ
     power = ctypes.c_uint()
@@ -414,12 +394,12 @@ def hlmlDeviceGetPowerManagementDefaultLimit(device: hlml_t.HLML_DEVICE.TYPE) ->
     check_return(ret)
     return power.value
 
-def hlmlDeviceGetECCMode(device: hlml_t.HLML_DEVICE.TYPE) -> dict:
-    """ Retrieves the current and pending ECC modes of the device
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            out (dict) - The current and pending ECC modes for the device.
+def hlmlDeviceGetECCMode(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.hlml_ecc_mode:
+    """Get current and pending ECC modes of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        hlml_ecc_mode: ECC mode info with 'current' and 'pending' attributes.
     """
     global _hlmlOBJ
     out = hlml_t.hlml_ecc_mode()
@@ -436,14 +416,13 @@ def hlmlDeviceGetECCMode(device: hlml_t.HLML_DEVICE.TYPE) -> dict:
     return out
 
 def hlmlDeviceGetTotalECCErrors(device: hlml_t.HLML_DEVICE.TYPE, error_type: hlml_t.HLML_MEMORY_ERROR.TYPE, counter_type: hlml_t.HLML_ECC_COUNTER) -> int:
-    """ Returns the number of ECC errors for a device. Number is from the last reset, or driver
-        reinstall. Currently only the number of corrected errors is supported.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            error_type ( 0-CORRECTED / 1-UNCORRECTED ) - The type of errors to count.
-            counter_type ( 0-VOLATILE / 1-AGGREGATE ) - The type of counter to use.
-        Returns:
-            count (int) - The number of ECC errors for the device, specified by parameters.
+    """Get number of ECC errors for a device since last reset or driver reinstall.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        error_type (HLML_MEMORY_ERROR.TYPE): Error type to count.
+        counter_type (HLML_ECC_COUNTER): Counter type to use.
+    Returns:
+        int: Number of ECC errors.
     """
     global _hlmlOBJ
     count = ctypes.c_longlong()
@@ -455,16 +434,16 @@ def hlmlDeviceGetTotalECCErrors(device: hlml_t.HLML_DEVICE.TYPE, error_type: hlm
     return count.value
 
 def hlmlDeviceGetMemoryErrorCounter(device: hlml_t.HLML_DEVICE.TYPE, error_type: hlml_t.HLML_MEMORY_ERROR.TYPE, counter_type: hlml_t.HLML_ECC_COUNTER.TYPE, location: hlml_t.HLML_MEMORY_LOCATION.TYPE) -> int:
-    """ Returns the number of ECC errors for a device at a specified memory location.
-        Number is from the last reset, or driver reinstall. Currently only the number
-        of corrected errors is supported.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            error_type ( 0-CORRECTED / 1-UNCORRECTED ) - The type of errors to count.
-            counter_type ( 0-VOLATILE / 1-AGGREGATE ) - The type of counter to use.
-            location ( 0-SRAM / 1-DRAM ) - The type of memory.
-        Returns:
-            count (int) - The number of ECC errors for the device, specified by parameters.
+    """Get number of ECC errors for a device at a specified memory location.
+        The number is from the last reset or driver reinstall. Currently, only the
+        number of corrected errors is supported.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        error_type (HLML_MEMORY_ERROR.TYPE): Error type to count.
+        counter_type (HLML_ECC_COUNTER.TYPE): Counter type to use.
+        location (HLML_MEMORY_LOCATION.TYPE): Memory location.
+    Returns:
+        int: Number of ECC errors.
     """
     global _hlmlOBJ
     ecc_count = ctypes.c_longlong()
@@ -475,12 +454,12 @@ def hlmlDeviceGetMemoryErrorCounter(device: hlml_t.HLML_DEVICE.TYPE, error_type:
     check_return(ret)
     return ecc_count.value
 
-def hlmlDeviceGetUUID(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Returns the UUID of the device
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            name (str) - The UUID for the given device.
+def hlmlDeviceGetUUID(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get UUID (Universal Unique ID) of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Device UUID.
     """
     global _hlmlOBJ
     name_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -493,12 +472,12 @@ def hlmlDeviceGetUUID(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     return name.value
 
 def hlmlDeviceGetMinorNumber(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves the minor number of the device ( maps to the device node file )
-        at /sys/class/accel/accel[minor]
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            number (int) - The minor number for the device.
+    """Get minor number of the device (maps to device node file
+        at /sys/class/accel/accel[minor]).
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Minor number.
     """
     global _hlmlOBJ
     number = ctypes.c_uint()
@@ -510,10 +489,9 @@ def hlmlDeviceGetMinorNumber(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return number.value
 
 def hlmlEventSetCreate() -> hlml_t.HLML_EVENT_SET.TYPE:
-    """ Create an empty set of events
-        Parameters: None.
-        Returns:
-            st (HLML_EVENT_SET) - An empty set of events.
+    """Create an empty set of events.
+    Returns:
+        HLML_EVENT_SET.TYPE: Empty event set.
     """
     global _hlmlOBJ
     st = hlml_t.HLML_EVENT_SET.TYPE()
@@ -525,10 +503,9 @@ def hlmlEventSetCreate() -> hlml_t.HLML_EVENT_SET.TYPE:
     return st
 
 def hlmlEventSetFree(st: hlml_t.HLML_EVENT_SET.TYPE) -> None:
-    """ Release a set of events
-        Parameters:
-            st (HLML_EVENT_SET) - The set of events to be released.
-        Returns: None.
+    """Release a set of events.
+    Args:
+        st (HLML_EVENT_SET.TYPE): Event set to release.
     """
     global _hlmlOBJ
 
@@ -542,12 +519,11 @@ def hlmlDeviceRegisterEvents(
         device: hlml_t.HLML_DEVICE.TYPE, event_types: int,
         st: hlml_t.HLML_EVENT_SET.TYPE
     ) -> None:
-    """ Start recording events on input device add events to input set
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            event_types ( 0-ECC_err / 1-Crit_err / 2-Clock_change ) - The type of events to start recording.
-            st (HLML_EVENT_SET) - The set of events to be released.
-        Returns: None.
+    """Start recording events on device and add events to set.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        event_types (int): Event types to record.
+        st (HLML_EVENT_SET.TYPE): Event set (buffer) to be written to.
     """
     global _hlmlOBJ
 
@@ -558,14 +534,15 @@ def hlmlDeviceRegisterEvents(
     return None
 
 def hlmlEventSetWait(st: hlml_t.HLML_EVENT_SET.TYPE, timeout: int) -> hlml_t.c_hlml_event_data:
-    """ Waits on events and delivers events.
-        If some events are ready to be delivered at the time of the call, function returns immediately.
-        If there are no events ready to be delivered, function sleeps until the event arrives but not longer than the specified timeout.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            timeout (int) - The maximum time to wait for a new event.
-        Returns:
-            data (c_hlml_event_data) - The data from events ready to be delivered.
+    """Wait for events and deliver event data.
+        If any events are available at the time of the call, function returns immediately.
+        If there are no events available, it waits until an event arrives or the specified
+        timeout is reached.
+    Args:
+        st (HLML_EVENT_SET.TYPE): Event set (buffer) to be written to.
+        timeout (int): Maximum time to wait for an event.
+    Returns:
+        c_hlml_event_data: Event data.
     """
     global _hlmlOBJ
     data = hlml_t.c_hlml_event_data()
@@ -577,14 +554,14 @@ def hlmlEventSetWait(st: hlml_t.HLML_EVENT_SET.TYPE, timeout: int) -> hlml_t.c_h
     return data
 
 def hlmlDeviceGetMACInfo(
-        device: hlml_t.HLML_DEVICE.TYPE, count=20, start=1) -> hlml_t.c_hlml_mac_info:
-    """ Get MAC addresses of device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            Count (int) - How many addresses to return.
-            Start (int) - The index to start at.
-        Returns:
-            mac (c_hlml_mac_info array of size count) - The MAC addresses of the device.
+        device: hlml_t.HLML_DEVICE.TYPE, count=20, start=1) -> tuple:
+    """Get MAC addresses of device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        count (int): Number of addresses to return.
+        start (int): Index to start at.
+    Returns:
+        c_hlml_mac_info array: MAC addresses.
     """
     global _hlmlOBJ
     mac = (hlml_t.c_hlml_mac_info * count)()
@@ -597,11 +574,11 @@ def hlmlDeviceGetMACInfo(
     return mac
 
 def hlmlDeviceGetHLRevision(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Returns HL revision
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            rev (int) - The HL revision
+    """Get HL revision of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: HL revision.
     """
     global _hlmlOBJ
     rev = ctypes.c_int()
@@ -613,11 +590,11 @@ def hlmlDeviceGetHLRevision(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return rev.value
 
 def hlmlDeviceGetPCBInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_pcb_info:
-    """ Returns the PCB info
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            pcb (c_hlml_pcb_info) - The info about the device's PCB.
+    """Get PCB info of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        c_hlml_pcb_info: PCB info.
     """
     global _hlmlOBJ
     pcb = hlml_t.c_hlml_pcb_info()
@@ -628,12 +605,12 @@ def hlmlDeviceGetPCBInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_pcb_i
     check_return(ret)
     return pcb
 
-def hlmlDeviceGetSerial(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Returns the unique board sn
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            ser (str) - The serial number of the device.
+def hlmlDeviceGetSerial(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get unique board serial number.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Serial number.
     """
     global _hlmlOBJ
     ser_len = hlml_t.HLML_DEFINE.HL_FIELD_MAX_SIZE
@@ -646,11 +623,11 @@ def hlmlDeviceGetSerial(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     return ser.value
 
 def hlmlDeviceGetModuleID(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves the module id configured on the device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            module_id (int) - The module id configured on the device.
+    """Get module ID configured on the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Module ID.
     """
     global _hlmlOBJ
     module_id = ctypes.c_uint()
@@ -662,11 +639,11 @@ def hlmlDeviceGetModuleID(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return module_id.value
 
 def hlmlDeviceGetBoardID(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves the device boardID ( 0 - 7 )
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            brd (int) - The board id of which slot the device is in.
+    """Get device board ID (slot number between 0 and 7).
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Board ID.
     """
     global _hlmlOBJ
     brd = ctypes.c_uint()
@@ -678,13 +655,12 @@ def hlmlDeviceGetBoardID(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return brd.value
 
 def hlmlDeviceGetPCIEThroughput(device: hlml_t.HLML_DEVICE.TYPE, counter_type: int) -> int:
-    """ Retrieve PCIe utilization information ( over 10ms interval )
-        counter_type ( 0-TX / 1-RX )
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            counter_type (int) -----------------------------------------------------------------------------
-        Returns:
-            throughput (int) - The throughput on the PCIE Transfer or Recieve Connection.
+    """Get PCIe utilization information (over 10ms interval).
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        counter_type (int): 0 for TX, 1 for RX.
+    Returns:
+        int: PCIe throughput.
     """
     global _hlmlOBJ
     pcie = ctypes.c_uint()
@@ -696,11 +672,11 @@ def hlmlDeviceGetPCIEThroughput(device: hlml_t.HLML_DEVICE.TYPE, counter_type: i
     return pcie.value
 
 def hlmlDeviceGetPCIEReplayCounter(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieve the PCIe replay counter
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            pcie (int) - The replay counter of the PCIE device.
+    """Get PCIe replay counter.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Replay counter.
     """
     global _hlmlOBJ
     pcie = ctypes.c_uint()
@@ -712,11 +688,11 @@ def hlmlDeviceGetPCIEReplayCounter(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return pcie.value
 
 def hlmlDeviceGetCurrPCIELinkGeneration(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieve the current PCIe link generation
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            link (int) - The generation of the device's PCIe link.
+    """Get current PCIe link generation.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: PCIe link generation.
     """
     global _hlmlOBJ
     link = ctypes.c_uint()
@@ -728,11 +704,11 @@ def hlmlDeviceGetCurrPCIELinkGeneration(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return link.value
 
 def hlmlDeviceGetCurrPCIELinkWidth(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieve the current PCIe link width
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            width (int) - The width or lanes of the PCIE connection.
+    """Get current PCIe link width.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: PCIe link width.
     """
     global _hlmlOBJ
     width = ctypes.c_uint()
@@ -744,11 +720,11 @@ def hlmlDeviceGetCurrPCIELinkWidth(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return width.value
 
 def hlmlDeviceGetCurrentClocksThrottleReasons(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves the current clocks reason for throttling
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            reason (int) - A code for the reason for the current clock throttle.
+    """Get current clocks throttle reason.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Throttle reason code.
     """
     global _hlmlOBJ
     reason = ctypes.c_ulonglong()
@@ -760,11 +736,11 @@ def hlmlDeviceGetCurrentClocksThrottleReasons(device: hlml_t.HLML_DEVICE.TYPE) -
     return reason.value
 
 def hlmlDeviceGetTotalEnergyConsumption(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Retrieves total energy consumption in mJ since the driver was last reloaded
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            energy (int) - Total energy consumption of the habana device.
+    """Get total energy consumption in mJ since last driver reload.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Total energy consumption in mJ.
     """
     global _hlmlOBJ
     energy = ctypes.c_ulonglong()
@@ -776,14 +752,13 @@ def hlmlDeviceGetTotalEnergyConsumption(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return energy.value
 
 def hlmlDeviceGetMacAddrInfo(device: hlml_t.HLML_DEVICE.TYPE) -> Tuple[Tuple[int, int], Tuple[int, int]]:
-    """ Retrieves the masks for supported ports and external ports.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            (mask, ext_mask)
-            mask is a tuple of two ints containing bitmask for supported ports.
-            mask_ext is a tuple of two ints containing bitmask for external
-                ports within the supported ports.
+    """Get masks for supported and external ports.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        Tuple[Tuple[int, int], Tuple[int, int]]: (mask, ext_mask), where:
+            mask: Bitmask of supported ports.
+            ext_mask: Bitmask of external ports (subset of supported).
     """
     global _hlmlOBJ
     mask = (ctypes.c_uint64 * 2)()
@@ -796,12 +771,12 @@ def hlmlDeviceGetMacAddrInfo(device: hlml_t.HLML_DEVICE.TYPE) -> Tuple[Tuple[int
     return (mask[0], mask[1]), (ext_mask[0], ext_mask[1])
 
 def hlmlDeviceNicGetLink(device: hlml_t.HLML_DEVICE.TYPE, port: int) -> bool:
-    """ Retrieves the NICs link status (up/down) for the requested port.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            port - Port for which the status is requested.
-        Returns:
-            up (bool) - Status of the port, True if up and False if down.
+    """Get NIC link status (up/down) for the requested port.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        port (int): Port number.
+    Returns:
+        bool: Port status. True if up, False if down.
     """
     global _hlmlOBJ
     up = ctypes.c_bool()
@@ -813,13 +788,13 @@ def hlmlDeviceNicGetLink(device: hlml_t.HLML_DEVICE.TYPE, port: int) -> bool:
     return up.value
 
 def hlmlDeviceNicGetStatistics(device: hlml_t.HLML_DEVICE.TYPE, port: int, num_of_counts: int = None) -> hlml_t.c_hlml_nic_stats_info:
-    """ Retrieves the NICs statistics for the requested port.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            port (int) - Port for which the stats is requested.
-            [optional] num_of_counts (int) - num of counts to allocate
-        Returns:
-            nic_stats_info (c_hlml_nic_stats_info) - The nic statistics
+    """Get NIC statistics for the requested port.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        port (int): Port number.
+        num_of_counts (int, optional): Number of counts to allocate.
+    Returns:
+        c_hlml_nic_stats_info: NIC statistics.
     """
     global _hlmlOBJ
     nic_stats_info = hlml_t.c_hlml_nic_stats_info(port, num_of_counts)
@@ -830,10 +805,10 @@ def hlmlDeviceNicGetStatistics(device: hlml_t.HLML_DEVICE.TYPE, port: int, num_o
     check_return(ret)
     return nic_stats_info
 
-def hlmlGetHLMLVersion() -> str:
-    """ Returns version of hlml library.
-        Returns:
-            ver (str) - Version of hlml library.
+def hlmlGetHLMLVersion() -> bytes:
+    """Get version of HLML library.
+    Returns:
+        bytes: HLML library version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -845,10 +820,10 @@ def hlmlGetHLMLVersion() -> str:
     check_return(ret)
     return version.value
 
-def hlmlGetDriverVersion() -> str:
-    """ Return version of driver.
-        Returns:
-            ver (str) - Version of driver.
+def hlmlGetDriverVersion() -> bytes:
+    """Get version of driver.
+    Returns:
+        bytes: Driver version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -860,10 +835,10 @@ def hlmlGetDriverVersion() -> str:
     check_return(ret)
     return version.value
 
-def hlmlGetNicDriverVersion() -> str:
-    """ Return version of NIC driver.
-        Returns:
-            ver (str) - Version of NIC driver.
+def hlmlGetNicDriverVersion() -> bytes:
+    """Get version of NIC driver.
+    Returns:
+        bytes: NIC driver version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -875,12 +850,12 @@ def hlmlGetNicDriverVersion() -> str:
     check_return(ret)
     return version.value
 
-def hlmlDeviceGetModelNumber(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Returns model number.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            model (str) - model number (in string format).
+def hlmlDeviceGetModelNumber(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get model number of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Model number.
     """
     global _hlmlOBJ
     num_len = hlml_t.HLML_DEFINE.HL_FIELD_MAX_SIZE
@@ -892,12 +867,12 @@ def hlmlDeviceGetModelNumber(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     check_return(ret)
     return num_str.value
 
-def hlmlDeviceGetFirmwareFITVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Return firmware fit version.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            ver (str) - firmware fit version.
+def hlmlDeviceGetFirmwareFITVersion(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get firmware FIT version.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Firmware FIT version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -909,12 +884,12 @@ def hlmlDeviceGetFirmwareFITVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     check_return(ret)
     return ver_str.value
 
-def hlmlDeviceGetFirmwareSPIVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Return firmware spi version.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            ver (str) - firmware spi version.
+def hlmlDeviceGetFirmwareSPIVersion(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get firmware SPI version.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Firmware SPI version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -926,12 +901,12 @@ def hlmlDeviceGetFirmwareSPIVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     check_return(ret)
     return ver_str.value
 
-def hlmlDeviceGetFWBootVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Return fw boot version.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            ver (str) - fw boot version.
+def hlmlDeviceGetFWBootVersion(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get firmware boot version.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Firmware boot version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -943,12 +918,12 @@ def hlmlDeviceGetFWBootVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     check_return(ret)
     return ver_str.value
 
-def hlmlDeviceGetFWOSVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Return fw os version.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            ver (str) - fw os version.
+def hlmlDeviceGetFWOSVersion(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get firmware OS version.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: Firmware OS version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -960,12 +935,12 @@ def hlmlDeviceGetFWOSVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     check_return(ret)
     return ver_str.value
 
-def hlmlDeviceGetCPLDVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Return cpld version.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            ver (str) - cpld version.
+def hlmlDeviceGetCPLDVersion(device: hlml_t.HLML_DEVICE.TYPE) -> bytes:
+    """Get CPLD version.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        bytes: CPLD version.
     """
     global _hlmlOBJ
     ver_len = hlml_t.COMMON_DEFINE.VERSION_MAX_LEN
@@ -978,11 +953,9 @@ def hlmlDeviceGetCPLDVersion(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     return ver_str.value
 
 def hlmlDeviceClearCpuAffinity(device: hlml_t.HLML_DEVICE.TYPE) -> None:
-    """ Clears a devices Cpu affinity
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            None.
+    """Clear device's CPU affinity.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
     """
     global _hlmlOBJ
 
@@ -993,12 +966,12 @@ def hlmlDeviceClearCpuAffinity(device: hlml_t.HLML_DEVICE.TYPE) -> None:
     return None
 
 def hlmlDeviceGetCpuAffinity(device: hlml_t.HLML_DEVICE.TYPE, cpu_set_size: int):
-    """ Retrieves the CPU affinity set associated with a device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            cpu_set_size - The size of the cpu set.
-        Returns:
-            cpu_set (int) - The cpu set.
+    """Get CPU affinity set associated with a device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        cpu_set_size (int): Size of the CPU set.
+    Returns:
+        array: CPU set.
     """
     global _hlmlOBJ
     cpu_set = (ctypes.c_ulong * cpu_set_size)()
@@ -1010,13 +983,13 @@ def hlmlDeviceGetCpuAffinity(device: hlml_t.HLML_DEVICE.TYPE, cpu_set_size: int)
     return cpu_set
 
 def hlmlDeviceGetCpuAffinityWithinScope(device: hlml_t.HLML_DEVICE.TYPE, cpu_set_size: int, scope: hlml_t.HLML_AFFINITY_SCOPE.TYPE):
-    """ Retrieves the CPU affinity set associated with a device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            cpu_set_size - The size of the cpu set.
-            scope - The affinity scope.
-        Returns:
-            cpu_set (int) - The cpu set.
+    """Get CPU affinity set associated with a device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        cpu_set_size (int): Size of the CPU set.
+        scope (HLML_AFFINITY_SCOPE.TYPE): Affinity scope.
+    Returns:
+        array: CPU set.
     """
     global _hlmlOBJ
     cpu_set = (ctypes.c_ulong * cpu_set_size)()
@@ -1028,13 +1001,13 @@ def hlmlDeviceGetCpuAffinityWithinScope(device: hlml_t.HLML_DEVICE.TYPE, cpu_set
     return cpu_set
 
 def hlmlDeviceGetMemoryAffinity(device: hlml_t.HLML_DEVICE.TYPE, node_set_size: int, scope: hlml_t.HLML_AFFINITY_SCOPE.TYPE):
-    """ Retrieves the memory affinity set associated with a device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            node_set_size - The size of the node set.
-            scope - The affinity scope.
-        Returns:
-            node_set (int) - The node set.
+    """Get memory affinity set associated with a device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        node_set_size (int): Size of the node set.
+        scope (HLML_AFFINITY_SCOPE.TYPE): Affinity scope.
+    Returns:
+        array: Node set.
     """
     global _hlmlOBJ
     node_set = (ctypes.c_ulong * node_set_size)()
@@ -1046,11 +1019,9 @@ def hlmlDeviceGetMemoryAffinity(device: hlml_t.HLML_DEVICE.TYPE, node_set_size:
     return node_set
 
 def hlmlDeviceSetCpuAffinity(device: hlml_t.HLML_DEVICE.TYPE) -> None:
-    """ Sets the CPU affinity with a device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            None.
+    """Set CPU affinity for a device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
     """
     global _hlmlOBJ
 
@@ -1061,14 +1032,12 @@ def hlmlDeviceSetCpuAffinity(device: hlml_t.HLML_DEVICE.TYPE) -> None:
     return None
 
 def hlmlDeviceGetViolationStatus(device: hlml_t.HLML_DEVICE.TYPE, perf_policy: hlml_t.HLML_PERF_POLICY.TYPE) -> hlml_t.c_hlml_violation_time:
-    """ Gets the violation status of a device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            perf_policy_type (HLML_PERF_POLICY.TYPE) - The perf policy type (
-               0=HLML_PERF_POLICY_POWER
-               1=HLML_PERF_POLICY_THERMAL)
-        Returns:
-            violation_time (c_hlml_violation_time) - The violation status of the device.
+    """Get violation status of a device for a given performance policy.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        perf_policy (HLML_PERF_POLICY.TYPE): Performance policy type.
+    Returns:
+        c_hlml_violation_time: Violation status, comprised of reference time and violation time.
     """
     global _hlmlOBJ
 
@@ -1081,12 +1050,12 @@ def hlmlDeviceGetViolationStatus(device: hlml_t.HLML_DEVICE.TYPE, perf_policy: h
     return violation_time
 
 def hlmlDeviceGetReplacedRowsCount(device: hlml_t.HLML_DEVICE.TYPE, cause: hlml_t.HLML_ROW_REPLACEMENT_CAUSE.TYPE) -> int:
-    """ Returns the number of replaced rows.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            cause  (HLML_ROW_REPLACEMENT_CAUSE.TYPE) - The replacement cause to query.
-        Returns:
-            row_count -> The number of replaced rows
+    """Get number of replaced rows for a given cause.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        cause (HLML_ROW_REPLACEMENT_CAUSE.TYPE): Replacement cause.
+    Returns:
+        int: Number of replaced rows.
     """
     global _hlmlOBJ
 
@@ -1099,12 +1068,13 @@ def hlmlDeviceGetReplacedRowsCount(device: hlml_t.HLML_DEVICE.TYPE, cause: hlml_
     return row_count.value
 
 def hlmlDeviceGetReplacedRows(device: hlml_t.HLML_DEVICE.TYPE, cause: hlml_t.HLML_ROW_REPLACEMENT_CAUSE.TYPE, row_count: int):
-    """ Returns an array of rows replaced by the given cause.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-            cause  (HLML_ROW_REPLACEMENT_CAUSE.TYPE) - The replacement cause to query.
-        Returns:
-            addresses - An array of hlml_t.c_hlml_row_address structures.
+    """Get array of replaced rows for a given cause.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+        cause (HLML_ROW_REPLACEMENT_CAUSE.TYPE): Replacement cause.
+        row_count (int): Number of rows to retrieve.
+    Returns:
+        array: Array of hlml_t.c_hlml_row_address structures.
     """
     global _hlmlOBJ
 
@@ -1118,11 +1088,11 @@ def hlmlDeviceGetReplacedRows(device: hlml_t.HLML_DEVICE.TYPE, cause: hlml_t.HLM
     return addresses
 
 def hlmlDeviceGetReplacedRowsPendingStatus(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Returns the pending status of replaced rows.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            is_pending - If there is pending row status.
+    """Get pending status of replaced rows.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Pending status. True if there are pending replaced rows, False otherwise.
     """
     global _hlmlOBJ
 
@@ -1135,11 +1105,11 @@ def hlmlDeviceGetReplacedRowsPendingStatus(device: hlml_t.HLML_DEVICE.TYPE) -> i
     return is_pending.value
 
 def hlmlDeviceGetOperationStatus(device: hlml_t.HLML_DEVICE.TYPE) -> str:
-    """ Return Operation Status.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            status (str) - operation status.
+    """Get operation status of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        str: Operation status.
     """
     global _hlmlOBJ
     sts_len = hlml_t.COMMON_DEFINE.STATUS_MAX_LEN
@@ -1151,12 +1121,12 @@ def hlmlDeviceGetOperationStatus(device: hlml_t.HLML_DEVICE.TYPE) -> str:
     check_return(ret)
     return sts_str.value
 
-def hlmlDeviceGetPowerManagementMode(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLML_ENABLE_STATE:
-    """ Acquire name of device from handle
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            mode (int) - The power management mode of the device.
+def hlmlDeviceGetPowerManagementMode(device: hlml_t.HLML_DEVICE.TYPE) -> int:
+    """Get power management mode of the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Power management mode.
     """
     global _hlmlOBJ
     mode = ctypes.c_uint()
@@ -1167,12 +1137,11 @@ def hlmlDeviceGetPowerManagementMode(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.
     return mode.value
 
 def hlmlDeviceGetPowerManagementLimit(device: hlml_t.HLML_DEVICE.TYPE) -> int:
-    """ Returns the pending status of replaced rows.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            limit (int) - Reference in which to return the power management
-                          limit in mW
+    """Get power management limit of the device in mW.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        int: Power management limit in mW.
     """
     global _hlmlOBJ
     power_max = ctypes.c_uint()
@@ -1183,12 +1152,11 @@ def hlmlDeviceGetPowerManagementLimit(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     return power_max.value
 
 def hlmlDeviceGetPowerManagementLimitConstraints(device: hlml_t.HLML_DEVICE.TYPE) -> Tuple[int, int]:
-    """ Returns the power management limit constraints of the device.
-        Parameters:
-            device (HLML_DEVICE.TYPE) - The handle for a habana device.
-        Returns:
-            min_limit (int) - The minimum power management limit of the device.
-            max_limit (int) - The maximum power management limit of the device.
+    """Get power management limit constraints (min, max) for the device.
+    Args:
+        device (HLML_DEVICE.TYPE): Device handle.
+    Returns:
+        Tuple[int, int]: (min_limit, max_limit)
     """
     global _hlmlOBJ
     min_limit = ctypes.c_uint()
@@ -1199,4 +1167,3 @@ def hlmlDeviceGetPowerManagementLimitConstraints(device: hlml_t.HLML_DEVICE.TYPE
 
     check_return(ret)
     return min_limit.value, max_limit.value
-