femtocrux 3.0.0__py3-none-any.whl → 3.2.0__py3-none-any.whl

femtocrux/VERSION

@@ -1 +1 @@
-3.0.0
+3.2.0

femtocrux/client/client.py

@@ -21,7 +21,8 @@ from femtocrux.util.utils import (
     read_secret_raw,
     get_channel_options,
     serialize_sim_inputs_message,
-    deserialize_simulation_output,
+    deserialize_simulation_data,
+    deserialize_simulation_data_list,
 )
 
 from google.protobuf.message import Message
@@ -274,9 +275,9 @@ class Simulator:
 
     def simulate(
         self,
-        inputs: Iterable[np.array],
+        inputs: list[dict[str, np.ndarray]] | dict[str, np.ndarray],
         input_period: float = 0.016,
-    ) -> List[np.array]:
+    ) -> tuple:
         """
         Simulates the model on the given inputs.
 
@@ -285,11 +286,28 @@ class Simulator:
                 "input_name1": np.ndarray(int16 types ...),
                 "input_name2": np.ndarray([int16 types ...])
             }
+            # Array has dimensions for sequence, features
+
+            or a list of dicts
+            [
+                {"input_name1": np.ndarray(int16 types ..),
+                 "input_name2": np.ndarray(int16 ...)},
+                {"input_name1": np.ndarray(int16 types ..),
+                 "input_name2": np.ndarray(int16 ...)},
+                {"input_name1": np.ndarray(int16 types ..),
+                 "input_name2": np.ndarray(int16 ...)},
+            ]
+
+            The sequence dimension is captured as list elements and the
+            array only has a feature dimension
+
         :type input_period (float, optional): Duration between each input in a sequence,
             in seconds.
 
-        :rtype: list
-        :return: The output tensors.
+        :rtype: tuple
+        :return: The deserialzed outputs either as a
+                   list[str, np.ndarray.shape(features)] or
+                   dict[str, np.ndarray.shape(sequence, features))] and the sim report.
 
         """
 
@@ -299,7 +317,16 @@ class Simulator:
         self._send_request(simulation_request)
         response = self._get_response()
 
-        return deserialize_simulation_output(response.outputs), response.report
+        which = response.WhichOneof("output_data")
+        if which == "outputs":
+            deserialized_outputs = deserialize_simulation_data(response.outputs)
+        elif which == "list_outputs":
+            deserialized_outputs = deserialize_simulation_data_list(
+                response.list_outputs
+            )
+        else:
+            raise (Exception("Didn't get output data as proto dict or list message."))
+        return deserialized_outputs, response.report
 
 
 class CompilerClientImpl:

femtocrux/grpc/compiler_service_pb2.py

@@ -26,17 +26,15 @@ from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
 from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x16\x63ompiler_service.proto\x12\nfscompiler\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"g\n\x04\x66qir\x12\r\n\x05model\x18\x01 \x01(\x0c\x12\x16\n\tbatch_dim\x18\x02 \x01(\x03H\x00\x88\x01\x01\x12\x19\n\x0csequence_dim\x18\x03 \x01(\x03H\x01\x88\x01\x01\x42\x0c\n\n_batch_dimB\x0f\n\r_sequence_dim\"G\n\x06tflite\x12\r\n\x05model\x18\x01 \x01(\x0c\x12\x1b\n\x0esignature_name\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x11\n\x0f_signature_name\"\xd0\x02\n\x12model_and_metadata\x12 \n\x04\x66qir\x18\x01 \x01(\x0b\x32\x10.fscompiler.fqirH\x00\x12$\n\x06tflite\x18\x02 \x01(\x0b\x32\x12.fscompiler.tfliteH\x00\x12\x13\n\x06iospec\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x18\n\x0bopt_profile\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x38\n\x12user_configuration\x18\x05 \x01(\x0b\x32\x17.google.protobuf.StructH\x03\x88\x01\x01\x12\x39\n\x13memory_reservations\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructH\x04\x88\x01\x01\x42\x04\n\x02irB\t\n\x07_iospecB\x0e\n\x0c_opt_profileB\x15\n\x13_user_configurationB\x16\n\x14_memory_reservations\"&\n\x06status\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x0b\n\x03msg\x18\x02 \x01(\t\"\xbc\x01\n\x12\x63ompiled_artifacts\x12\x0f\n\x07\x62itfile\x18\x01 \x01(\x0c\x12\x13\n\x06iospec\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x39\n\x13memory_reservations\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructH\x01\x88\x01\x01\x12\"\n\x06status\x18\x04 \x01(\x0b\x32\x12.fscompiler.statusB\t\n\x07_iospecB\x16\n\x14_memory_reservations\"5\n\x07ndarray\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\x12\r\n\x05shape\x18\x02 \x03(\x03\x12\r\n\x05\x64type\x18\x03 \x01(\t\"\x14\n\x04\x64\x61ta\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"\xe6\x01\n\x0fsimulation_data\x12\x37\n\x06inputs\x18\x01 \x03(\x0b\x32\'.fscompiler.simulation_data.InputsEntry\x12\x19\n\x0csim_duration\x18\x02 \x01(\x02H\x00\x88\x01\x01\x12\x19\n\x0cinput_period\x18\x03 \x01(\x02H\x01\x88\x01\x01\x1a\x42\n\x0bInputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.fscompiler.ndarray:\x02\x38\x01\x42\x0f\n\r_sim_durationB\x0f\n\r_input_period\"\x81\x01\n\x10simulation_input\x12/\n\x05model\x18\x01 \x01(\x0b\x32\x1e.fscompiler.model_and_metadataH\x00\x12+\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x1b.fscompiler.simulation_dataH\x00\x42\x0f\n\rmodel_or_data\"\xc9\x01\n\x11simulation_output\x12;\n\x07outputs\x18\x01 \x03(\x0b\x32*.fscompiler.simulation_output.OutputsEntry\x12\x0e\n\x06report\x18\x02 \x01(\t\x12\"\n\x06status\x18\x03 \x01(\x0b\x32\x12.fscompiler.status\x1a\x43\n\x0cOutputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.fscompiler.ndarray:\x02\x38\x01\"\x1f\n\x0cversion_info\x12\x0f\n\x07version\x18\x01 \x01(\t2\x92\x02\n\x07\x43ompile\x12K\n\x07\x63ompile\x12\x1e.fscompiler.model_and_metadata\x1a\x1e.fscompiler.compiled_artifacts\"\x00\x12,\n\x04ping\x12\x10.fscompiler.data\x1a\x10.fscompiler.data\"\x00\x12M\n\x08simulate\x12\x1c.fscompiler.simulation_input\x1a\x1d.fscompiler.simulation_output\"\x00(\x01\x30\x01\x12=\n\x07version\x12\x16.google.protobuf.Empty\x1a\x18.fscompiler.version_info\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x16\x63ompiler_service.proto\x12\nfscompiler\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"g\n\x04\x66qir\x12\r\n\x05model\x18\x01 \x01(\x0c\x12\x16\n\tbatch_dim\x18\x02 \x01(\x03H\x00\x88\x01\x01\x12\x19\n\x0csequence_dim\x18\x03 \x01(\x03H\x01\x88\x01\x01\x42\x0c\n\n_batch_dimB\x0f\n\r_sequence_dim\"G\n\x06tflite\x12\r\n\x05model\x18\x01 \x01(\x0c\x12\x1b\n\x0esignature_name\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x11\n\x0f_signature_name\"\xd0\x02\n\x12model_and_metadata\x12 \n\x04\x66qir\x18\x01 \x01(\x0b\x32\x10.fscompiler.fqirH\x00\x12$\n\x06tflite\x18\x02 \x01(\x0b\x32\x12.fscompiler.tfliteH\x00\x12\x13\n\x06iospec\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x18\n\x0bopt_profile\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x38\n\x12user_configuration\x18\x05 \x01(\x0b\x32\x17.google.protobuf.StructH\x03\x88\x01\x01\x12\x39\n\x13memory_reservations\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructH\x04\x88\x01\x01\x42\x04\n\x02irB\t\n\x07_iospecB\x0e\n\x0c_opt_profileB\x15\n\x13_user_configurationB\x16\n\x14_memory_reservations\"&\n\x06status\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x0b\n\x03msg\x18\x02 \x01(\t\"\xbc\x01\n\x12\x63ompiled_artifacts\x12\x0f\n\x07\x62itfile\x18\x01 \x01(\x0c\x12\x13\n\x06iospec\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x39\n\x13memory_reservations\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructH\x01\x88\x01\x01\x12\"\n\x06status\x18\x04 \x01(\x0b\x32\x12.fscompiler.statusB\t\n\x07_iospecB\x16\n\x14_memory_reservations\"5\n\x07ndarray\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\x12\r\n\x05shape\x18\x02 \x03(\x03\x12\r\n\x05\x64type\x18\x03 \x01(\t\"\x14\n\x04\x64\x61ta\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"\xdb\x01\n\x0fsimulation_data\x12+\n\x06inputs\x18\x01 \x01(\x0b\x32\x19.fscompiler.str_array_mapH\x00\x12\x35\n\x0blist_inputs\x18\x02 \x01(\x0b\x32\x1e.fscompiler.list_str_array_mapH\x00\x12\x19\n\x0csim_duration\x18\x03 \x01(\x02H\x01\x88\x01\x01\x12\x19\n\x0cinput_period\x18\x04 \x01(\x02H\x02\x88\x01\x01\x42\x0c\n\ninput_dataB\x0f\n\r_sim_durationB\x0f\n\r_input_period\"\x84\x01\n\rstr_array_map\x12\x31\n\x04\x64\x61ta\x18\x01 \x03(\x0b\x32#.fscompiler.str_array_map.DataEntry\x1a@\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.fscompiler.ndarray:\x02\x38\x01\"=\n\x12list_str_array_map\x12\'\n\x04maps\x18\x01 \x03(\x0b\x32\x19.fscompiler.str_array_map\"\x81\x01\n\x10simulation_input\x12/\n\x05model\x18\x01 \x01(\x0b\x32\x1e.fscompiler.model_and_metadataH\x00\x12+\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x1b.fscompiler.simulation_dataH\x00\x42\x0f\n\rmodel_or_data\"\xbc\x01\n\x11simulation_output\x12,\n\x07outputs\x18\x01 \x01(\x0b\x32\x19.fscompiler.str_array_mapH\x00\x12\x36\n\x0clist_outputs\x18\x02 \x01(\x0b\x32\x1e.fscompiler.list_str_array_mapH\x00\x12\x0e\n\x06report\x18\x03 \x01(\t\x12\"\n\x06status\x18\x04 \x01(\x0b\x32\x12.fscompiler.statusB\r\n\x0boutput_data\"\x1f\n\x0cversion_info\x12\x0f\n\x07version\x18\x01 \x01(\t2\x92\x02\n\x07\x43ompile\x12K\n\x07\x63ompile\x12\x1e.fscompiler.model_and_metadata\x1a\x1e.fscompiler.compiled_artifacts\"\x00\x12,\n\x04ping\x12\x10.fscompiler.data\x1a\x10.fscompiler.data\"\x00\x12M\n\x08simulate\x12\x1c.fscompiler.simulation_input\x1a\x1d.fscompiler.simulation_output\"\x00(\x01\x30\x01\x12=\n\x07version\x12\x16.google.protobuf.Empty\x1a\x18.fscompiler.version_info\"\x00\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'compiler_service_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
   DESCRIPTOR._loaded_options = None
-  _globals['_SIMULATION_DATA_INPUTSENTRY']._loaded_options = None
-  _globals['_SIMULATION_DATA_INPUTSENTRY']._serialized_options = b'8\001'
-  _globals['_SIMULATION_OUTPUT_OUTPUTSENTRY']._loaded_options = None
-  _globals['_SIMULATION_OUTPUT_OUTPUTSENTRY']._serialized_options = b'8\001'
+  _globals['_STR_ARRAY_MAP_DATAENTRY']._loaded_options = None
+  _globals['_STR_ARRAY_MAP_DATAENTRY']._serialized_options = b'8\001'
   _globals['_FQIR']._serialized_start=97
   _globals['_FQIR']._serialized_end=200
   _globals['_TFLITE']._serialized_start=202
@@ -52,17 +50,19 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_DATA']._serialized_start=900
   _globals['_DATA']._serialized_end=920
   _globals['_SIMULATION_DATA']._serialized_start=923
-  _globals['_SIMULATION_DATA']._serialized_end=1153
-  _globals['_SIMULATION_DATA_INPUTSENTRY']._serialized_start=1053
-  _globals['_SIMULATION_DATA_INPUTSENTRY']._serialized_end=1119
-  _globals['_SIMULATION_INPUT']._serialized_start=1156
-  _globals['_SIMULATION_INPUT']._serialized_end=1285
-  _globals['_SIMULATION_OUTPUT']._serialized_start=1288
-  _globals['_SIMULATION_OUTPUT']._serialized_end=1489
-  _globals['_SIMULATION_OUTPUT_OUTPUTSENTRY']._serialized_start=1422
-  _globals['_SIMULATION_OUTPUT_OUTPUTSENTRY']._serialized_end=1489
-  _globals['_VERSION_INFO']._serialized_start=1491
-  _globals['_VERSION_INFO']._serialized_end=1522
-  _globals['_COMPILE']._serialized_start=1525
-  _globals['_COMPILE']._serialized_end=1799
+  _globals['_SIMULATION_DATA']._serialized_end=1142
+  _globals['_STR_ARRAY_MAP']._serialized_start=1145
+  _globals['_STR_ARRAY_MAP']._serialized_end=1277
+  _globals['_STR_ARRAY_MAP_DATAENTRY']._serialized_start=1213
+  _globals['_STR_ARRAY_MAP_DATAENTRY']._serialized_end=1277
+  _globals['_LIST_STR_ARRAY_MAP']._serialized_start=1279
+  _globals['_LIST_STR_ARRAY_MAP']._serialized_end=1340
+  _globals['_SIMULATION_INPUT']._serialized_start=1343
+  _globals['_SIMULATION_INPUT']._serialized_end=1472
+  _globals['_SIMULATION_OUTPUT']._serialized_start=1475
+  _globals['_SIMULATION_OUTPUT']._serialized_end=1663
+  _globals['_VERSION_INFO']._serialized_start=1665
+  _globals['_VERSION_INFO']._serialized_end=1696
+  _globals['_COMPILE']._serialized_start=1699
+  _globals['_COMPILE']._serialized_end=1973
 # @@protoc_insertion_point(module_scope)

femtocrux/server/compiler_frontend.py

@@ -51,7 +51,7 @@ class CompilerFrontend:
         self,
         input_ir: GraphProto,
         iospec: IOSpec | None = None,
-        opt_profile: Literal["efficient", "fast", "conservative"] | None = "efficient",
+        opt_profile: Literal["efficient", "fast", "conservative"] | None = None,
         user_configuration: dict | None = None,
         memory_reservations: dict | None = None,
     ) -> tuple[FASMIR, IOSpec, dict]:
@@ -100,7 +100,7 @@ class CompilerFrontend:
 
     def run_behavioral_simulator(
         self,
-        inputs: dict[str, np.ndarray],
+        inputs: dict[str, np.ndarray] | list[dict[str, np.ndarray]],
         input_period: float = None,
         iospec: IOSpec | None = None,
         **kwargs
@@ -135,7 +135,7 @@ class TorchCompiler(CompilerFrontend):
         self,
         input_ir: GraphProto,
         iospec: IOSpec | None = None,
-        opt_profile: Literal["efficient", "fast", "conservative"] | None = "efficient",
+        opt_profile: Literal["efficient", "fast", "conservative"] | None = None,
         user_configuration: dict | None = None,
         memory_reservations: dict | None = None,
     ) -> tuple[FASMIR, IOSpec, dict]:

femtocrux/server/server.py

@@ -15,6 +15,7 @@ from femtocrux.util.utils import (
     get_channel_options,
     serialize_simulation_output,
     deserialize_simulation_data,
+    deserialize_simulation_data_list,
 )
 
 # Import GRPC artifacts
@@ -165,7 +166,16 @@ class CompileServicer(cs_pb2_grpc.CompileServicer):
             try:
                 if data_request.WhichOneof("model_or_data") == "data":
                     ### New Path with int64 proto message holding data
-                    deserialized_inputs = deserialize_simulation_data(data.inputs)
+
+                    which = data.WhichOneof("input_data")
+                    if which == "inputs":
+                        deserialized_inputs = deserialize_simulation_data(data.inputs)
+                    elif which == "list_inputs":
+                        deserialized_inputs = deserialize_simulation_data_list(
+                            data.list_inputs
+                        )
+                    else:
+                        raise (Exception("Didn't get input data as dict or list."))
                     outputs, metrics = compiler.run_behavioral_simulator(
                         deserialized_inputs,
                         input_period=field_or_none(data, "input_period"),

femtocrux/util/utils.py

@@ -31,47 +31,108 @@ def serialize_numpy_array(arr: np.ndarray) -> cs_pb2.ndarray:
     )
 
 
+def prepare_array_for_serialization(array, type_of_input="dict"):
+    """
+    Check that the arrays that are sent for inputs have the correct types and shapes.
+
+    They have to be np.ndarrays and have the appropriate shapes:
+    List[Dict[str, np.ndarray]] the ndarray has to only have 1 dimension of features
+    Dict[str, np.ndarray] the ndarray can have 2 dimensions, one for sequence and
+    another for features.
+    """
+    if isinstance(array, torch.Tensor):
+        array_to_serialize = array.numpy()
+    elif isinstance(array, np.ndarray):
+        array_to_serialize = array
+    else:
+        raise (Exception("Input array was not of type torch.Tensor or np.ndarray"))
+
+    if not np.issubdtype(array_to_serialize.dtype, np.integer):
+        raise (
+            Exception(
+                "Input data is not an integer type. Please quantize your"
+                "data to int16 or lower."
+            )
+        )
+    shape_size = 2
+    if type_of_input == "dict":
+        shape_size = 2
+    elif type_of_input == "list":
+        shape_size = 1
+
+    if len(array_to_serialize.shape) > shape_size:
+        raise (
+            Exception(
+                f"Expected {shape_size} dimensions for input and got shape: "
+                f"{array_to_serialize.shape} Your input array has too many "
+                f"dimensions. When in inference mode, please remove any batch "
+                f"dimensions."
+            )
+        )
+    return array_to_serialize
+
+
 def serialize_sim_inputs_message(
-    data_dict: dict, input_period: float
+    input_data: list | dict, input_period: float
 ) -> cs_pb2.simulation_data:
-    """Serializes a dictionary where values are NumPy arrays."""
+    """
+    Serialize input_data which could be the list[dict[str, np.ndarray]] or
+    dict[str, np.ndarray] format of input into oneof.
+
+    list_str_array_map or str_array_map proto messages and create a larger
+    simulation_data proto message.
+
+    return simulation_data proto message
+    """
     message = cs_pb2.simulation_data()
-    for key, array in data_dict.items():
-        if isinstance(array, torch.Tensor):
-            array_to_serialize = array.numpy()
-        elif isinstance(array, np.ndarray):
-            array_to_serialize = array
-        else:
-            raise (Exception("Input array was not of type torch.Tensor or np.ndarray"))
-
-        if not np.issubdtype(array_to_serialize.dtype, np.integer):
-            raise (
-                Exception(
-                    "Input data is not an integer type. Please quantize your"
-                    "data to int16 or lower."
-                )
-            )
 
-        if len(array_to_serialize.shape) > 2:
-            raise (
-                Exception(
-                    "Expected 2 dimensions for input and got shape: "
-                    "{array_to_serialize.shape} Your input array has too many "
-                    "dimensions. When in inference mode, please remove any batch "
-                    "dimensions."
+    if isinstance(input_data, list):
+        for single_entry in input_data:
+            input_map = message.list_inputs.maps.add()
+            for key, array in single_entry.items():
+                array_to_serialize = prepare_array_for_serialization(
+                    array, type_of_input="list"
                 )
-            )
+                input_map.data[key].CopyFrom(serialize_numpy_array(array_to_serialize))
 
-        message.inputs[key].CopyFrom(serialize_numpy_array(array_to_serialize))
+    elif isinstance(input_data, dict):
+        for key, array in input_data.items():
+            array_to_serialize = prepare_array_for_serialization(
+                array, type_of_input="dict"
+            )
+            message.inputs.data[key].CopyFrom(serialize_numpy_array(array_to_serialize))
+    else:
+        raise (
+            Exception(f"input_data was of type {type(input_data)}, not dict or list.")
+        )
     message.input_period = input_period
     return message
 
 
-def serialize_simulation_output(data_dict: dict, report) -> cs_pb2.simulation_output:
-    """Serializes a dictionary where values are NumPy arrays."""
+def serialize_simulation_output(
+    output_data: dict | list, report
+) -> cs_pb2.simulation_output:
+    """
+    Serialize the output_data and report into a proto message simulation_output.
+
+    We have both the list[dict[str, np.ndarray]] and the dict[str, np.ndarray]
+    as possible output formats.
+
+    We serialize whichever one we get as message.list_outputs or message.outputs
+
+    return the proto message
+    """
     message = cs_pb2.simulation_output()
-    for key, array in data_dict.items():
-        message.outputs[key].CopyFrom(serialize_numpy_array(array))
+
+    if isinstance(output_data, list):
+        for single_entry in output_data:
+            output_map = message.list_outputs.maps.add()
+            for key, array in single_entry.items():
+                output_map.data[key].CopyFrom(serialize_numpy_array(array))
+    elif isinstance(output_data, dict):
+        for key, array in output_data.items():
+            message.outputs.data[key].CopyFrom(serialize_numpy_array(array))
+
     message.report = json.dumps(report)
     message.status.CopyFrom(cs_pb2.status(success=True))
 
@@ -83,14 +144,23 @@ def deserialize_numpy_array(proto: cs_pb2.ndarray) -> np.ndarray:
     return np.frombuffer(proto.data, dtype=np.dtype(proto.dtype)).reshape(proto.shape)
 
 
-def deserialize_simulation_data(proto: cs_pb2.simulation_data) -> dict:
-    """Deserializes a LargerMessage back into a dictionary of NumPy arrays."""
-    return {key: deserialize_numpy_array(value) for key, value in proto.items()}
-
-
-def deserialize_simulation_output(proto: cs_pb2.simulation_output) -> dict:
+def deserialize_simulation_data(proto: cs_pb2.str_array_map) -> dict:
     """Deserializes a LargerMessage back into a dictionary of NumPy arrays."""
-    return {key: deserialize_numpy_array(value) for key, value in proto.items()}
+    return {key: deserialize_numpy_array(value) for key, value in proto.data.items()}
+
+
+def deserialize_simulation_data_list(list_input: cs_pb2.list_str_array_map) -> list:
+    """
+    Convert proto list_str_array_map into a python list[dict[str, np.ndarray]]
+    """
+    list_of_dicts = []
+    for input_map in list_input.maps:
+        dict_entry = {}
+        for key, ndarray_proto in input_map.data.items():
+            # Convert ndarray proto to whatever format you need
+            dict_entry[key] = deserialize_numpy_array(ndarray_proto)
+        list_of_dicts.append(dict_entry)
+    return list_of_dicts
 
 
 def read_secret_raw(prompt="Secret: "):

{femtocrux-3.0.0.dist-info → femtocrux-3.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: femtocrux
-Version: 3.0.0
+Version: 3.2.0
 Summary: Femtosense Compiler
 Home-page: https://github.com/femtosense/femtocrux
 Author: Femtosense

{femtocrux-3.0.0.dist-info → femtocrux-3.2.0.dist-info}/RECORD

@@ -1,22 +1,22 @@
 femtocrux/ENV_REQUIREMENTS.sh,sha256=t_O1B4hJAMgxvH9gwp1qls6eVFmhSYBJe64KmuK_H-4,1389
 femtocrux/PY_REQUIREMENTS,sha256=UwXV0o3gieruZUYdN9r2bqQ0Wcf_vosjeP6LJVx6oF0,275
-femtocrux/VERSION,sha256=KYW-iyjTrehY6Nj7S8IvVlsb9gIN_5gtzhQfdyG5mZw,6
+femtocrux/VERSION,sha256=M72vni1ry94uTmE-hl5AP4_iH8YhB400wwsL3H9o2QM,6
 femtocrux/__init__.py,sha256=VeuEHuCvQmqHgzj_ogjIbwGE_E8UFLYSJN6GTpXBfe0,409
 femtocrux/version.py,sha256=uNg2kHxQo6oUN1ah7s9_85rCZVRoTHGPD1GAQPZW4lw,164
 femtocrux/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-femtocrux/client/client.py,sha256=UrO8OYhw6KTbI_tHz_DpRwBx93MwO1MvrkMUnAQaR94,29797
+femtocrux/client/client.py,sha256=QgyNoLdbjWbGyznWqQQyjJOgGGfGVdgiWxz-297H5Pw,31005
 femtocrux/grpc/__init__.py,sha256=uiMHQt5I2eAKJqI3Zh0h1Gm7cmPR4PbaGS71nCJQCGw,169
-femtocrux/grpc/compiler_service_pb2.py,sha256=ChHlC3TvWNFR-umRg2l7j3FAiybe1EGBEJfYB0Z78zM,6084
+femtocrux/grpc/compiler_service_pb2.py,sha256=a95hVO0IOLTuDmmag-acQaW6v8FFVAsoOkB6_8hw6vo,6255
 femtocrux/grpc/compiler_service_pb2_grpc.py,sha256=b_fyld4rpKE4n5MimKOoyNZ5VU4UqUS_R3EM3_daqxM,8539
 femtocrux/server/__init__.py,sha256=kpm2AGLxo3AA9zo-G8N0Dm0gmWbvAd6jfhMbtkeHcxo,77
-femtocrux/server/compiler_frontend.py,sha256=kXVg_JIQyLY9zdwT1s5xusWaBd-9t4gDwXrU6FX6UqI,6217
+femtocrux/server/compiler_frontend.py,sha256=4m7Sk8eVdFMvpS4U3fqdDd_QpNv-n1lnoxuqH2_LdHE,6233
 femtocrux/server/exceptions.py,sha256=lI6n471n5QKf5G3aL_1kuBVEItD-jBgithVVpPDwNYc,609
 femtocrux/server/healthcheck.py,sha256=ehqAwnv0D0zpy-AUZAPwv8rp874DZCwUmP8nzdXzZvI,1565
-femtocrux/server/server.py,sha256=k7GiLHxtzo5UH1O0edav4yEPgYD682DmB1aTZodtBuQ,8268
+femtocrux/server/server.py,sha256=102uwMQQHGdNm9GXDqAHaKjDlpYRFkWxDl-7NOwcwP4,8721
 femtocrux/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-femtocrux/util/utils.py,sha256=v8v09aDaOtfejpSIB-LgaA--JJj9o2_BWvNwjd8wfww,4134
-femtocrux-3.0.0.dist-info/licenses/LICENSE,sha256=eN9ZI1xHjUmFvN3TEeop5kBGXRUBfbsl55KBNBYYFqI,36
-femtocrux-3.0.0.dist-info/METADATA,sha256=TZkC3EUSy5ZqBYmeGccdq4HAYx3NL_wQoYWipcKccDk,2763
-femtocrux-3.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-femtocrux-3.0.0.dist-info/top_level.txt,sha256=BkTttlioC3je__8577wxRieZqY3Abu7FOOdMnmYbcNI,10
-femtocrux-3.0.0.dist-info/RECORD,,
+femtocrux/util/utils.py,sha256=esE9DIT1__9smSmehhBGCY0dxLvequ2RXGXqfH4R24s,6491
+femtocrux-3.2.0.dist-info/licenses/LICENSE,sha256=eN9ZI1xHjUmFvN3TEeop5kBGXRUBfbsl55KBNBYYFqI,36
+femtocrux-3.2.0.dist-info/METADATA,sha256=8dsm6SO9xFvURsKigS93XpRRho37noQoGY-iQqr_Jkg,2763
+femtocrux-3.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+femtocrux-3.2.0.dist-info/top_level.txt,sha256=BkTttlioC3je__8577wxRieZqY3Abu7FOOdMnmYbcNI,10
+femtocrux-3.2.0.dist-info/RECORD,,