opengradient 0.3.23__py3-none-any.whl → 0.3.25__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.23.dist-info → opengradient-0.3.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: opengradient
-Version: 0.3.23
+Version: 0.3.25
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License

opengradient-0.3.25.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.25.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.3.25.dist-info/METADATA,sha256=hX9dG02z0mVY5wKqwR48upLDje_24e_gGtsQb0Ry1IU,8754
+opengradient-0.3.25.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+opengradient-0.3.25.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.3.25.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.3.25.dist-info/RECORD,,

opengradient-0.3.23.dist-info/RECORD

@@ -1,26 +0,0 @@
-opengradient/__init__.py,sha256=uU_Onxf-4JMQ9gHMQPbJDbU93kbB8qNayBLrnZnRFzg,12742
-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.23.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.3.23.dist-info/METADATA,sha256=EhqCC5WEHRypug84TG0flHH-MsMDKKRN9B5D2DydaJY,8754
-opengradient-0.3.23.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-opengradient-0.3.23.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.3.23.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.3.23.dist-info/RECORD,,