Fixed lint and merge issues

Author: ysingh7

src/lava/magma/runtime/message_infrastructure/multiprocessing.py

@@ -19,8 +19,6 @@
 import multiprocessing as mp
 import traceback
 
-from lava.magma.compiler.channels.interfaces import ChannelType, Channel
-
 try:
     from lava.magma.core.model.c.type import LavaTypeTransfer
 except ImportError:

src/lava/magma/runtime/runtime.py

@@ -16,14 +16,15 @@
                                                        SupportTempChannel,
                                                        Selector,
                                                        getTempSendPort,
-                                                       getTempRecvPort)
+                                                       getTempRecvPort,
+                                                       AbstractTransferPort)
 
 from lava.magma.compiler.var_model import AbstractVarModel, LoihiSynapseVarModel
 from lava.magma.runtime.message_infrastructure.message_interface_enum import \
     ActorType
 from lava.magma.runtime.message_infrastructure.factory import \
     MessageInfrastructureFactory
-from lava.magma.runtime.\
+from lava.magma.runtime. \
     message_infrastructure.message_infrastructure_interface import \
     MessageInfrastructureInterface
 from lava.magma.runtime.mgmt_token_enums import (MGMT_COMMAND, MGMT_RESPONSE,
@@ -132,8 +133,7 @@ def __init__(self,
         self._req_stop: bool = False
         self.runtime_to_service: ty.Iterable[SendPort] = []
         self.service_to_runtime: ty.Iterable[RecvPort] = []
-        self._open_ports: ty.List[AbstractCspPort] = []
-
+        self._open_ports: ty.List[AbstractTransferPort] = []
 
     def __del__(self):
         """On destruction, terminate Runtime automatically to
@@ -462,11 +462,6 @@ def set_var(self, var_id: int, value: np.ndarray, idx: np.ndarray = None):
             buffer: np.ndarray = value
             if idx:
                 buffer = buffer[idx]
-            buffer_shape: ty.Tuple[int, ...] = buffer.shape
-            num_items: int = np.prod(buffer_shape).item()
-            reshape_order = 'F' if isinstance(
-                ev, LoihiSynapseVarModel) else 'C'
-            buffer = buffer.reshape((1, num_items), order=reshape_order)
 
             if SupportTempChannel:
                 addr_path = rsp_port.recv()
@@ -543,10 +538,7 @@ def get_var(self, var_id: int, idx: np.ndarray = None) -> np.ndarray:
                 reshape_order = 'F' if isinstance(ev, LoihiSynapseVarModel) \
                     else 'C'
                 buffer = buffer.reshape(ev.shape, order=reshape_order)
-            # 3. Reshape result and return
-            reshape_order = 'F' if isinstance(
-                ev, LoihiSynapseVarModel) else 'C'
-            buffer = buffer.reshape(ev.shape, order=reshape_order)
+
             if idx:
                 return buffer[idx]
             else:

tests/lava/magma/compiler/builders/test_builder.py

@@ -31,7 +31,6 @@
 )
 
 
-
 class MockMessageInterface:
     def __init__(self, smm):
         self.smm = smm

tests/lava/magma/compiler/test_channel_builder.py

@@ -22,7 +22,6 @@
 )
 
 
-
 class MockMessageInterface:
     def __init__(self, smm):
         self.smm = smm