Fix formatting

Author: lukamac

DeeployTest/testDmas.py

@@ -1,9 +1,10 @@
-from enum import Enum
 import itertools
 import subprocess
 from typing import Tuple
 
-def test(dma: str, inputShape: Tuple[int, ...], tileShape: Tuple[int, ...], nodeCount: int, dataType: str, doublebuffer: bool):
+
+def test(dma: str, inputShape: Tuple[int, ...], tileShape: Tuple[int, ...], nodeCount: int, dataType: str,
+         doublebuffer: bool):
     cfg_str = f"""
     - input shape: {inputShape}
     - tile shape: {tileShape}
@@ -34,6 +35,7 @@ def test(dma: str, inputShape: Tuple[int, ...], tileShape: Tuple[int, ...], node
         print(f"Rerun with command:\n{full_cmd}")
         exit(-1)
 
+
 # input shape, tile shape, node count, data type
 test_shapes_and_more = [
     ((10, 10), (10, 10), 1, "uint8_t"),

DeeployTest/testRunner_SnitchDma.py

@@ -1,5 +1,7 @@
 import numpy as np
 from testUtils.codeGenerate import generateTestNetwork
+from testUtils.dmaUtils import MemcpyLayer, MemcpyParser, MemcpyTileConstraint, MemcpyTypeChecker, generate_graph, \
+    memcpyTemplate, prepare_deployer_with_custom_tiling, setup_snitch_deployer
 from testUtils.testRunner import TestRunner, TestRunnerArgumentParser
 from testUtils.typeMapping import baseTypeFromName, dtypeFromDeeployType
 
@@ -15,8 +17,6 @@
 from Deeploy.Targets.Snitch.SnitchDma import SnitchDma
 from Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement import TilingVariableReplacement
 from Deeploy.TilingExtension.TilerExtension import TilingReadyNodeBindings
-from testUtils.dmaUtils import MemcpyLayer, MemcpyParser, MemcpyTileConstraint, MemcpyTypeChecker, \
-    generate_graph, memcpyTemplate, prepare_deployer_with_custom_tiling, setup_snitch_deployer
 
 testRunnerArgumentParser = TestRunnerArgumentParser(tiling_arguments = True)
 testRunnerArgumentParser.add_argument('--input-shape',

DeeployTest/testUtils/dmaUtils.py

@@ -9,7 +9,8 @@
 from Deeploy.DeeployTypes import NetworkContext, NetworkDeployer, NodeParser, NodeTemplate, NodeTypeChecker, \
     ONNXLayer, OperatorRepresentation, VariableBuffer
 from Deeploy.MemoryLevelExtension.MemoryLevels import MemoryHierarchy, MemoryLevel
-from Deeploy.MemoryLevelExtension.NetworkDeployers.MemoryLevelDeployer import MemoryDeployerWrapper, MemoryPlatformWrapper
+from Deeploy.MemoryLevelExtension.NetworkDeployers.MemoryLevelDeployer import MemoryDeployerWrapper, \
+    MemoryPlatformWrapper
 from Deeploy.MemoryLevelExtension.OptimizationPasses.MemoryLevelAnnotationPasses import AnnotateDefaultMemoryLevel, \
     AnnotateIOMemoryLevel
 from Deeploy.Targets.PULPOpen.Deployer import PULPDeployer