opengradient 0.3.24__py3-none-any.whl → 0.3.26__py3-none-any.whl

opengradient/utils.py

@@ -1,7 +1,7 @@
+import json
 import logging
 from decimal import Decimal
 from typing import Dict, List, Tuple
-import json
 
 import numpy as np
 from web3.datastructures import AttributeDict
@@ -17,20 +17,21 @@ def convert_to_fixed_point(number: float) -> Tuple[int, int]:
     # Converting number to string in case this is a numpy float
     decimal_val = Decimal(str(number)).normalize()
     sign, digits, exponent = decimal_val.as_tuple()
-    value = int(''.join(map(str, digits)))
+    value = int("".join(map(str, digits)))
     if sign:
         value = -value
-    
+
     if exponent >= 0:
-        value *= 10 ** exponent
+        value *= 10**exponent
         decimals = 0
     else:
         decimals = -exponent
-    
+
     logging.debug(f"Converted number {number} to fixed point value={value} decimals={decimals}")
     logging.debug(f"Types value={type(value)} decimals={type(decimals)}")
     return value, decimals
 
+
 def convert_to_float32(value: int, decimals: int) -> np.float32:
     """
     Converts fixed point back into floating point
@@ -39,9 +40,8 @@ def convert_to_float32(value: int, decimals: int) -> np.float32:
     """
     return np.float32(Decimal(value) / (10 ** Decimal(decimals)))
 
-def convert_to_model_input(
-        inputs: Dict[str, np.ndarray]
-    ) -> Tuple[List[Tuple[str, List[Tuple[int, int]]]], List[Tuple[str, List[str]]]]:
+
+def convert_to_model_input(inputs: Dict[str, np.ndarray]) -> Tuple[List[Tuple[str, List[Tuple[int, int]]]], List[Tuple[str, List[str]]]]:
     """
     Expect SDK input to be a dict with the format
         key: tensor name
@@ -66,9 +66,7 @@ def convert_to_model_input(
 
         # Check if type is np array
         if not isinstance(tensor_data, np.ndarray):
-            raise TypeError(
-                "Inference input must be list, numpy array, or type (str, int, float): %s" 
-                % type(tensor_data))
+            raise TypeError("Inference input must be list, numpy array, or type (str, int, float): %s" % type(tensor_data))
 
         # Flatten list and retain shape
         shape = tensor_data.shape
@@ -78,16 +76,16 @@ def convert_to_model_input(
         # Parse into number and string tensors
         if issubclass(tensor_data.dtype.type, np.floating):
             # Convert to fixed-point tuples
-            data_type = np.dtype([('value', int), ('decimal', int)])
+            data_type = np.dtype([("value", int), ("decimal", int)])
             converted_tensor_data = np.array([convert_to_fixed_point(i) for i in flat_data], dtype=data_type)
-            
+
             input = (tensor_name, converted_tensor_data.tolist(), shape)
             logging.debug("\tFloating tensor input: %s", input)
 
             number_tensors.append(input)
         elif issubclass(tensor_data.dtype.type, np.integer):
             # Convert to fixed-point tuples
-            data_type = np.dtype([('value', int), ('decimal', int)])
+            data_type = np.dtype([("value", int), ("decimal", int)])
             converted_tensor_data = np.array([convert_to_fixed_point(int(i)) for i in flat_data], dtype=data_type)
 
             input = (tensor_name, converted_tensor_data.tolist(), shape)
@@ -102,16 +100,17 @@ def convert_to_model_input(
             string_tensors.append(input)
         else:
             raise TypeError(f"Data type {tensor_data.dtype.type} not recognized")
-        
+
     logging.debug("Number tensors: %s", number_tensors)
     logging.debug("Number tensor types: %s", [type(item) for item in number_tensors])
     logging.debug("String tensors: %s", string_tensors)
     logging.debug("String tensor types: %s", [type(item) for item in string_tensors])
     return number_tensors, string_tensors
 
+
 def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
     """
-    Converts inference output into a user-readable output. 
+    Converts inference output into a user-readable output.
     Expects the inference node to return a dict with the format:
         key: output_name (str)
         value: (output_array (list), shape (list)) (tuple)
@@ -120,24 +119,24 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
     back into its original shape.
     """
     logging.debug(f"Parsing event data: {event_data}")
-        
+
     output_dict = {}
 
-    output = event_data.get('output', {})
+    output = event_data.get("output", {})
     logging.debug(f"Output data: {output}")
 
     if isinstance(output, AttributeDict):
         # Parse numbers
-        for tensor in output.get('numbers', []):
+        for tensor in output.get("numbers", []):
             logging.debug(f"Processing number tensor: {tensor}")
             if isinstance(tensor, AttributeDict):
-                name = tensor.get('name')
-                shape = tensor.get('shape')
+                name = tensor.get("name")
+                shape = tensor.get("shape")
                 values = []
                 # Convert from fixed point back into np.float32
-                for v in tensor.get('values', []):
+                for v in tensor.get("values", []):
                     if isinstance(v, AttributeDict):
-                        values.append(convert_to_float32(value=int(v.get('value')), decimals=int(v.get('decimals'))))
+                        values.append(convert_to_float32(value=int(v.get("value")), decimals=int(v.get("decimals"))))
                     else:
                         logging.warning(f"Unexpected number type: {type(v)}")
                 output_dict[name] = np.array(values).reshape(shape)
@@ -145,22 +144,22 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
                 logging.warning(f"Unexpected tensor type: {type(tensor)}")
 
         # Parse strings
-        for tensor in output.get('strings', []):
+        for tensor in output.get("strings", []):
             logging.debug(f"Processing string tensor: {tensor}")
             if isinstance(tensor, AttributeDict):
-                name = tensor.get('name')
-                shape = tensor.get('shape')
-                values = tensor.get('values', [])
+                name = tensor.get("name")
+                shape = tensor.get("shape")
+                values = tensor.get("values", [])
                 output_dict[name] = np.array(values).reshape(shape)
             else:
                 logging.warning(f"Unexpected tensor type: {type(tensor)}")
 
         # Parse JSON dicts
-        for tensor in output.get('jsons', []):
+        for tensor in output.get("jsons", []):
             logging.debug(f"Processing JSON tensor: {tensor}")
             if isinstance(tensor, AttributeDict):
-                name = tensor.get('name')
-                value = tensor.get('value')
+                name = tensor.get("name")
+                value = tensor.get("value")
                 output_dict[name] = np.array(json.loads(value))
             else:
                 logging.warning(f"Unexpected tensor type: {type(tensor)}")
@@ -170,4 +169,4 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
 
     logging.debug(f"Parsed output: {output_dict}")
 
-    return output_dict
+    return output_dict

{opengradient-0.3.24.dist-info → opengradient-0.3.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: opengradient
-Version: 0.3.24
+Version: 0.3.26
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License
@@ -35,103 +35,16 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: aiohappyeyeballs>=2.4.3
-Requires-Dist: aiohttp>=3.10.8
-Requires-Dist: aiosignal>=1.3.1
-Requires-Dist: annotated-types>=0.7.0
-Requires-Dist: attrs>=24.2.0
-Requires-Dist: bitarray>=2.9.2
-Requires-Dist: CacheControl>=0.14.0
-Requires-Dist: cachetools>=5.5.0
-Requires-Dist: certifi>=2024.8.30
-Requires-Dist: cffi>=1.17.1
-Requires-Dist: charset-normalizer>=3.3.2
-Requires-Dist: ckzg>=2.0.1
-Requires-Dist: cleo>=2.1.0
-Requires-Dist: click>=8.1.7
-Requires-Dist: cramjam>=2.8.4
-Requires-Dist: crashtest>=0.4.1
-Requires-Dist: cryptography>=43.0.1
-Requires-Dist: cytoolz>=0.12.3
-Requires-Dist: distlib>=0.3.8
-Requires-Dist: dulwich>=0.21.7
 Requires-Dist: eth-account>=0.13.4
-Requires-Dist: eth-hash>=0.7.0
-Requires-Dist: eth-keyfile>=0.8.1
-Requires-Dist: eth-keys>=0.5.1
-Requires-Dist: eth-rlp>=2.1.0
-Requires-Dist: eth-typing>=5.0.0
-Requires-Dist: eth-utils>=5.0.0
-Requires-Dist: eth_abi>=5.1.0
-Requires-Dist: fastjsonschema>=2.20.0
-Requires-Dist: fastparquet>=2024.5.0
-Requires-Dist: filelock>=3.16.1
+Requires-Dist: web3>=7.3.0
+Requires-Dist: click>=8.1.7
 Requires-Dist: firebase-rest-api>=1.11.0
-Requires-Dist: frozenlist>=1.4.1
-Requires-Dist: fsspec>=2024.9.0
-Requires-Dist: google-api-core>=2.20.0
-Requires-Dist: google-auth>=2.35.0
-Requires-Dist: google-cloud-core>=2.4.1
-Requires-Dist: google-cloud-firestore>=2.19.0
-Requires-Dist: google-cloud-storage>=2.18.2
-Requires-Dist: google-crc32c>=1.6.0
-Requires-Dist: google-resumable-media>=2.7.2
-Requires-Dist: googleapis-common-protos>=1.65.0
 Requires-Dist: grpcio>=1.66.2
-Requires-Dist: grpcio-tools>=1.66.2
-Requires-Dist: protobuf>=4.24.0
-Requires-Dist: hexbytes>=1.2.1
-Requires-Dist: idna>=3.10
-Requires-Dist: jaraco.classes>=3.4.0
-Requires-Dist: jwcrypto>=1.5.6
-Requires-Dist: keyring>=24.3.1
+Requires-Dist: numpy>=1.26.4
+Requires-Dist: requests>=2.32.3
 Requires-Dist: langchain>=0.3.7
-Requires-Dist: more-itertools>=10.5.0
-Requires-Dist: msgpack>=1.1.0
-Requires-Dist: multidict>=6.1.0
 Requires-Dist: openai>=1.58.1
-Requires-Dist: packaging>=24.1
-Requires-Dist: pandas>=2.2.3
-Requires-Dist: parsimonious>=0.10.0
-Requires-Dist: pathlib>=1.0.1
-Requires-Dist: pexpect>=4.9.0
-Requires-Dist: pkce>=1.0.3
-Requires-Dist: pkginfo>=1.11.1
-Requires-Dist: platformdirs>=4.3.6
-Requires-Dist: proto-plus>=1.24.0
-Requires-Dist: protobuf>=5.28.2
-Requires-Dist: ptyprocess>=0.7.0
-Requires-Dist: pyarrow>=17.0.0
-Requires-Dist: pyasn1>=0.6.1
-Requires-Dist: pyasn1_modules>=0.4.1
-Requires-Dist: pycparser>=2.22
-Requires-Dist: pycryptodome>=3.21.0
 Requires-Dist: pydantic>=2.9.2
-Requires-Dist: pydantic_core>=2.23.4
-Requires-Dist: pyproject_hooks>=1.2.0
-Requires-Dist: python-dateutil>=2.9.0.post0
-Requires-Dist: python-jwt>=4.1.0
-Requires-Dist: pytz>=2024.2
-Requires-Dist: pyunormalize>=16.0.0
-Requires-Dist: RapidFuzz>=3.10.0
-Requires-Dist: regex>=2024.9.11
-Requires-Dist: requests>=2.32.3
-Requires-Dist: requests-toolbelt>=1.0.0
-Requires-Dist: rlp>=4.0.1
-Requires-Dist: rsa>=4.9
-Requires-Dist: shellingham>=1.5.4
-Requires-Dist: six>=1.16.0
-Requires-Dist: tomlkit>=0.13.2
-Requires-Dist: toolz>=0.12.1
-Requires-Dist: trove-classifiers>=2024.9.12
-Requires-Dist: types-requests>=2.32.0.20240914
-Requires-Dist: typing_extensions>=4.12.2
-Requires-Dist: tzdata>=2024.2
-Requires-Dist: urllib3>=2.2.3
-Requires-Dist: web3>=7.3.0
-Requires-Dist: websockets>=13.1
-Requires-Dist: xattr>=1.1.0
-Requires-Dist: yarl>=1.13.1
 
 # OpenGradient Python SDK
 Python SDK for the OpenGradient platform provides decentralized model management & inference services. Python SDK allows programmatic access to our model repository and decentralized AI infrastructure. 

opengradient-0.3.26.dist-info/RECORD

@@ -0,0 +1,26 @@
+opengradient/__init__.py,sha256=j52gE9yLsig3og9_EKhV12K-VzFpZLORmM0msm2zJB0,12858
+opengradient/account.py,sha256=5wrYpws_1lozjOFjLCTHtxgoxK-LmObDAaVy9eDcJY4,1145
+opengradient/cli.py,sha256=uRvzICY_QqWor025-pQvovL4xuCmOSk0Z2uyHsJtS3g,25582
+opengradient/client.py,sha256=YmNrDjn9Zd71O0M_pgmnKQ3CqWFPfIDp0004UpRFt_A,41064
+opengradient/defaults.py,sha256=VK3a4B4d6ZpGRqfpMe20PTUMXACjH2TdbCkiWLld8WU,398
+opengradient/exceptions.py,sha256=88tfegboGtlehQcwhxsl6ZzhLJWZWlkf_bkHTiCtXpo,3391
+opengradient/types.py,sha256=rjnn6d7ugv4kO2tJw87QWiC4aBvei8renVw2NMVNhEo,3989
+opengradient/utils.py,sha256=hf1dQvOHdCFthrAr_Wif_PNn6-C3zZaa3QCZX1HMWoA,6911
+opengradient/abi/ModelExecutorHistorical.abi,sha256=Z-2CQRG2BO5R-Ix1OLgQD9GCcHZw8w6-Zth3rzNX0TU,3278
+opengradient/abi/inference.abi,sha256=MR5u9npZ-Yx2EqRW17_M-UnGgFF3mMEMepOwaZ-Bkgc,7040
+opengradient/contracts/templates/ModelExecutorHistorical.bin,sha256=J_bHglQOXEsv44BkBkMvA9uWz7c9c8gdqqSb0KxIzkI,34924
+opengradient/llm/__init__.py,sha256=b_msjZstmTRD20LOaZbBxxigtnL7vxh7CziiyVlqpAo,1104
+opengradient/llm/og_langchain.py,sha256=F9gdrwqMFPjLUakUfSjc4Obx8qIvf7HIphe6eM3ObZo,4322
+opengradient/llm/og_openai.py,sha256=9MA3HDotjcJbbcGJ7DYZUA990QHQLdWpxAYhJSsTiz0,3708
+opengradient/mltools/__init__.py,sha256=JU1VDzYb-rpko_hn9IlwjOvy_dfW1hvxBCBC2xrnvUM,147
+opengradient/mltools/model_tool.py,sha256=dNoznwAiauff961EzM_9raYGdwcPLIqxIsPE1yCuWPU,5086
+opengradient/proto/__init__.py,sha256=AhaSmrqV0TXGzCKaoPV8-XUvqs2fGAJBM2aOmDpkNbE,55
+opengradient/proto/infer.proto,sha256=13eaEMcppxkBF8yChptsX9HooWFwJKze7oLZNl-LEb8,1217
+opengradient/proto/infer_pb2.py,sha256=sGWDDVumYhXoCJTG9rLyvKu4XyaEjPE_b038kbNlj7w,3484
+opengradient/proto/infer_pb2_grpc.py,sha256=q42_eZ7OZCMTXdWocYA4Ka3B0c3B74dOhfqdaIOO5AU,6700
+opengradient-0.3.26.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.3.26.dist-info/METADATA,sha256=8emmrKaTCefGRESCgFIBx4AHmLaRNTD_V6HoO7odY3c,5833
+opengradient-0.3.26.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+opengradient-0.3.26.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.3.26.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.3.26.dist-info/RECORD,,

opengradient-0.3.24.dist-info/RECORD

@@ -1,26 +0,0 @@
-opengradient/__init__.py,sha256=tv3JB_R52vy0lHnnUioee6CJ3kjnQAriRHZ3iHY06aY,12741
-opengradient/account.py,sha256=2B7rtCXQDX-yF4U69h8B9-OUreJU4IqoGXG_1Hn9nWs,1150
-opengradient/cli.py,sha256=vYE_zr1oKYOkiAEWRyW3cR6jJQr6xKterjI1grhmROg,26399
-opengradient/client.py,sha256=Dybu7g5kP7rJKQq0NbNjamJUVNTYLdSFdEd3ZA6fiaQ,42668
-opengradient/defaults.py,sha256=6tsW9Z84z6YtITCsULTTgnN0KRUZjfSoeWJZqdWkYCo,384
-opengradient/exceptions.py,sha256=v4VmUGTvvtjhCZAhR24Ga42z3q-DzR1Y5zSqP_yn2Xk,3366
-opengradient/types.py,sha256=Z8HFmCxNiC4Pr8qVoALF_jYOAqIJuKsiJSgq96q-sxQ,4175
-opengradient/utils.py,sha256=lUDPmyPqLwpZI-owyN6Rm3QvUjOn5pLN5G1QyriVm-E,6994
-opengradient/abi/ModelExecutorHistorical.abi,sha256=Z-2CQRG2BO5R-Ix1OLgQD9GCcHZw8w6-Zth3rzNX0TU,3278
-opengradient/abi/inference.abi,sha256=MR5u9npZ-Yx2EqRW17_M-UnGgFF3mMEMepOwaZ-Bkgc,7040
-opengradient/contracts/templates/ModelExecutorHistorical.bin,sha256=J_bHglQOXEsv44BkBkMvA9uWz7c9c8gdqqSb0KxIzkI,34924
-opengradient/llm/__init__.py,sha256=J1W_AKPntqlDqLeflhn2x7A0i-dkMT-ol3jlEdFgMWU,1135
-opengradient/llm/og_langchain.py,sha256=-QBGFdQJ1cdSCTUUerajHXNvV5Xi9yaVpa3BeAqKZbQ,4541
-opengradient/llm/og_openai.py,sha256=3oaQOQnA7RzQ-LcFw-AQRn2RON8tsprSa9bGtyD88OE,3853
-opengradient/mltools/__init__.py,sha256=cfzf6lZmqLNN1efa5K_pCa7k1D5PnRkB6LNyJ9UeJvk,162
-opengradient/mltools/model_tool.py,sha256=q1k3Zq7jmRe-8HqP_-ia-p7ocDRcSn18T4gbNM9xlD0,5237
-opengradient/proto/__init__.py,sha256=AhaSmrqV0TXGzCKaoPV8-XUvqs2fGAJBM2aOmDpkNbE,55
-opengradient/proto/infer.proto,sha256=13eaEMcppxkBF8yChptsX9HooWFwJKze7oLZNl-LEb8,1217
-opengradient/proto/infer_pb2.py,sha256=wg2vjLQCNv6HRhYuIqgj9xivi3nO4IPz6E5wh2dhDqY,3446
-opengradient/proto/infer_pb2_grpc.py,sha256=y5GYwD1EdNs892xx58jdfyA0fO5QC7k3uZOtImTHMiE,6891
-opengradient-0.3.24.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.3.24.dist-info/METADATA,sha256=yy6OzS1b8s4ZXj1YICJNLA1EHv2Igg3Vyu8MG8CuRUs,8754
-opengradient-0.3.24.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-opengradient-0.3.24.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.3.24.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.3.24.dist-info/RECORD,,