purem 3.0.1__py3-none-any.whl → 3.0.2__py3-none-any.whl

purem/__init__.py

@@ -2,5 +2,4 @@ from purem.env_config import load_env_config
 
 from purem.core import Purem
 
-
-purem = Purem(licenced_key=load_env_config().PUREM_LICENSE_KEY)
+purem = Purem()

purem/core.py

@@ -11,7 +11,6 @@ Additional Use Grant: Free for personal and non-commercial research use only.
 SPDX-License-Identifier: BUSL-1.1
 """
 
-import base64
 import ctypes
 import json
 import os
@@ -19,17 +18,16 @@ import shutil
 import ssl
 import urllib.request
 import zipfile
-from typing import List, Optional, Dict
+from pathlib import Path
+from typing import Optional, Dict
 
 import certifi as certifi
-import numpy as np
 from numpy import ndarray
 
-from purem.env_config import load_env_config
+from purem.env_config import load_env_config, EnvConfig
 from purem.file_structure import FileStructure
 from purem.loader import Loader
 from purem.logger import Logger
-from purem.utils import _compute_shifted_jit
 
 
 class Purem:
@@ -68,7 +66,7 @@ class Purem:
     :type _log: Logger
     """
 
-    def __init__(self, licenced_key: Optional[str] = None):
+    def __init__(self):
         """
         Represents the initialization and configuration of an environment, license key,
         and file structure required for the system's binary operations.
@@ -76,26 +74,30 @@ class Purem:
         :param licenced_key: Optional license key string for initializing the system.
         :type licenced_key: Optional[str]
         """
-        self._license_key = licenced_key or None
-        self._lib = None
-        self._download_binary_url = None
+        self._lib: Optional[ctypes.CDLL] = None
+        self._download_binary_url: Optional[str] = None
         self._ctx = ssl.create_default_context(cafile=certifi.where())
         self._file_structure = FileStructure()
-        self._binary_path = self._file_structure.get_binary_path()
-        self._binary_project_root_path = (
+        self._binary_path: Path = self._file_structure.get_binary_path()
+        self._binary_project_root_path: Path = (
             self._file_structure.get_binary_project_root_path()
         )
-        self._binary_archive_path = self._file_structure.get_binary_archive_path()
-        self._binary_archive_path_tmp = (
+        self._binary_archive_path: Path = self._file_structure.get_binary_archive_path()
+        self._binary_archive_path_tmp: Path = (
             self._file_structure.get_binary_archive_path_tmp()
         )
-        self._env = load_env_config()
-        self._config_url = (
-            self._env.PUREM_CONFIG_URL
-            or "https://api.worktif.com/v2/portal/products/purem/config"
+        self._env: EnvConfig = load_env_config()
+        self._license_key: Optional[str] = self._env.PUREM_LICENSE_KEY or None
+        self._config_url: str = (
+                self._env.PUREM_CONFIG_URL
+                or "https://api.worktif.com/v2/portal/products/purem/config"
         )
-        self._loader = Loader()
-        self._log = Logger()
+        self._loader: Loader = Loader()
+        self._log: Logger = Logger()
+        print('self._license_key: ', self._license_key)
+
+        if self._license_key is not None:
+            self.configure(license_key=self._license_key)
 
     def configure(self, license_key: Optional[str] = None) -> None:
         """
@@ -125,9 +127,18 @@ class Purem:
         self._set_binary()
 
     def softmax(self, array: ndarray) -> ndarray:
-        ptr = array.ctypes.data_as(ctypes.POINTER(ctypes.c_float))
-        self._lib.purem(ptr, array.size)
-        return array
+        try:
+            ptr = array.ctypes.data_as(ctypes.POINTER(ctypes.c_float))
+            self._lib.purem(ptr, array.size)
+            return array
+        except Exception as e:
+            raise ValueError(
+                self._log.info(
+                    "Purem requires a valid license key to initialize.\n"
+                    "You can obtain your key at https://worktif.com or through your enterprise deployment.",
+                    verbose=True
+                )
+            )
 
     def softmax_pure(self, ptr, size) -> None:
         """
@@ -168,7 +179,7 @@ class Purem:
         binary_url = f"{protocol}://{base}/{path}{self._license_key}"
         return binary_url
 
-    def _tune_binary(self):
+    def _tune_binary(self) -> None:
         """
         Tunes the binary by loading the specified binary file as a shared library
         and setting up its function signatures for further use.
@@ -185,7 +196,7 @@ class Purem:
         self._lib.purem.argtypes = [ctypes.POINTER(ctypes.c_float), ctypes.c_size_t]
         self._lib.purem.restype = None
 
-    def _tune_project_root_binary(self):
+    def _tune_project_root_binary(self) -> None:
         """
         Tuning the project root binary configuration.
 
@@ -228,7 +239,7 @@ class Purem:
         except Exception:
             return None
 
-    def _set_binary(self):
+    def _set_binary(self) -> None:
         """
         Sets and initializes the binary for the Purem runtime environment. The method ensures that a valid binary
         is available and properly configured. If a valid license key or binary is not found, the method will attempt
@@ -262,8 +273,8 @@ class Purem:
                 )
                 self._loader.start()
                 self._download_binary_url = (
-                    self._build_url(self._load_from_latest_cdn_index())
-                    or f"{self._env.PUREM_DOWNLOAD_BINARY_URL}{self._license_key}"
+                        self._build_url(self._load_from_latest_cdn_index())
+                        or f"{self._env.PUREM_DOWNLOAD_BINARY_URL}{self._license_key}"
                 )
                 self._download_and_extract_binary()
                 self._loader.stop()
@@ -288,7 +299,7 @@ class Purem:
                     )
                 )
 
-    def _download_and_extract_binary(self):
+    def _download_and_extract_binary(self) -> None:
         """
         Downloads a binary file from a given URL, saves it temporarily, and extracts its
         contents to a specific directory. Handles errors related to incomplete or
@@ -337,3 +348,16 @@ class Purem:
             )
 
         self._binary_archive_path.unlink()
+
+    def has_license_key(self) -> bool:
+        """
+        Checks if an object has a valid license key.
+
+        This method evaluates whether the `_license_key` attribute of
+        the object is not `None`, implying the presence of a license key.
+
+        :return: Returns `True` if the `_license_key` attribute is not
+            `None`, otherwise returns `False`.
+        :rtype: bool
+        """
+        return self._license_key is not None

purem/logger.py

@@ -19,8 +19,8 @@ class Logger:
         self._env = load_env_config()
         self._default_info_message = f"Something went wrong"
 
-    def info(self, message: str) -> str:
-        if self._env.PUREM_VERBOSE is True:
+    def info(self, message: str, verbose: bool = False) -> str:
+        if self._env.PUREM_VERBOSE is True or verbose is True:
             return f"[purem]: {message}\n"
         else:
             return f"[purem]: {self._default_info_message}\n"

{purem-3.0.1.dist-info → purem-3.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: purem
-Version: 3.0.1
+Version: 3.0.2
 Summary: The official high-performance mapping function for mixed-type arrays powered by Work TIF Ltd.
 Author-email: Raman Marozau <raman@worktif.com>
 License-Expression: BUSL-1.1
@@ -25,7 +25,6 @@ Requires-Python: >=3.11
 Description-Content-Type: text/x-rst
 License-File: LICENSE
 Requires-Dist: numpy~=2.1.3
-Requires-Dist: pytest~=8.3.5
 Requires-Dist: numba~=0.61.0
 Requires-Dist: certifi~=2025.1.31
 Requires-Dist: pydantic~=2.10.6
@@ -33,6 +32,8 @@ Requires-Dist: dotenv~=0.9.9
 Requires-Dist: python-dotenv~=1.1.0
 Requires-Dist: setuptools~=65.5.1
 Requires-Dist: requests~=2.32.3
+Requires-Dist: pytest~=8.3.5
+Requires-Dist: pytest-env~=1.1.5
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: flake8; extra == "dev"

purem-3.0.2.dist-info/RECORD

@@ -0,0 +1,12 @@
+purem/__init__.py,sha256=of-z4lRvk7cmLLsCbiXHIPXVjP0FHBCByndtPgINfT0,92
+purem/core.py,sha256=kGhCPlSwyTlmjsv970H_DSCBrvMCL6TA2-EHmwsgQ3c,15830
+purem/env_config.py,sha256=o1VujOYEH0XXVHbM0j84BEbO2FqlWVQ088DxlSfsvRU,2663
+purem/file_structure.py,sha256=nYTSj74gDsnVAYAwkb5NGE1QHXvlkaYuaafB_FIAMaI,3337
+purem/loader.py,sha256=GDeuiqZcKvZQEDaUg2TO5rC-WwVcjIgKqNEpF602mmg,3610
+purem/logger.py,sha256=vZHUGwL-MXJPNBoHIxqLf26KOF9qk4xj2qdgUhRas8c,1281
+purem/utils.py,sha256=wjLHukHROX3GKJoEwXCR9X6CfqT70M1Mj6FcLtFUxJ8,923
+purem-3.0.2.dist-info/licenses/LICENSE,sha256=5WFXHK6Xc_wj2EtvzVtd9TIC4cWEZJS8ECNTuvutsiE,1636
+purem-3.0.2.dist-info/METADATA,sha256=fuDQr3LQrympk6wCLUqEbqb-Raxb5qxrAmrnH1BQZJg,5677
+purem-3.0.2.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+purem-3.0.2.dist-info/top_level.txt,sha256=EjS75KEpZUEKSV2TFGW6w5aLqY9nUyO6Gq2ATz-KeZM,6
+purem-3.0.2.dist-info/RECORD,,

purem-3.0.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-purem/__init__.py,sha256=zF_hOKYn8X3jeoBxQ9X1irDi6HCPQ-HRcMP5fcocaNw,141
-purem/core.py,sha256=fAWRKSzPp4x0YNKm_HrHWV9j38gISl2fLU6XyYIK33U,14772
-purem/env_config.py,sha256=o1VujOYEH0XXVHbM0j84BEbO2FqlWVQ088DxlSfsvRU,2663
-purem/file_structure.py,sha256=nYTSj74gDsnVAYAwkb5NGE1QHXvlkaYuaafB_FIAMaI,3337
-purem/loader.py,sha256=GDeuiqZcKvZQEDaUg2TO5rC-WwVcjIgKqNEpF602mmg,3610
-purem/logger.py,sha256=56pSIjyQal4IvaFzRpKBaQ8oMAdt-u638QK6uex-cRQ,1239
-purem/utils.py,sha256=wjLHukHROX3GKJoEwXCR9X6CfqT70M1Mj6FcLtFUxJ8,923
-purem-3.0.1.dist-info/licenses/LICENSE,sha256=5WFXHK6Xc_wj2EtvzVtd9TIC4cWEZJS8ECNTuvutsiE,1636
-purem-3.0.1.dist-info/METADATA,sha256=qe1GhJoHEKQyjhJH7NR4c4XVhNlPhifKA-H_bY8HITg,5644
-purem-3.0.1.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-purem-3.0.1.dist-info/top_level.txt,sha256=EjS75KEpZUEKSV2TFGW6w5aLqY9nUyO6Gq2ATz-KeZM,6
-purem-3.0.1.dist-info/RECORD,,