format

Author: calad0i

hls4ml/converters/onnx_to_hls.py

@@ -76,6 +76,7 @@ def get_input_shape(graph, node):
 def get_constant_value(graph, constant_name):
     tensor = next((x for x in graph.initializer if x.name == constant_name), None)
     from onnx import numpy_helper
+
     return numpy_helper.to_array(tensor)
 
 
@@ -274,6 +275,7 @@ def onnx_to_hls(config):
     print('Interpreting Model ...')
 
     import onnx
+
     onnx_model = onnx.load(config['OnnxModel']) if isinstance(config['OnnxModel'], str) else config['OnnxModel']
 
     layer_list, input_layers, output_layers = parse_onnx_model(onnx_model)

hls4ml/writer/oneapi_writer.py

@@ -102,9 +102,10 @@ def write_project_cpp(self, model):
         project_name = model.config.get_project_name()
 
         filedir = os.path.dirname(os.path.abspath(__file__))
-        with open(os.path.join(filedir, '../templates/oneapi/firmware/myproject.cpp')) as f, open(
-            f'{model.config.get_output_dir()}/src/firmware/{project_name}.cpp', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/firmware/myproject.cpp')) as f,
+            open(f'{model.config.get_output_dir()}/src/firmware/{project_name}.cpp', 'w') as fout,
+        ):
             model_inputs = model.get_input_variables()
             model_outputs = model.get_output_variables()
             model_brams = [var for var in model.get_weight_variables() if var.storage.lower() == 'bram']
@@ -207,9 +208,10 @@ def write_project_header(self, model):
         project_name = model.config.get_project_name()
 
         filedir = os.path.dirname(os.path.abspath(__file__))
-        with open(os.path.join(filedir, '../templates/oneapi/firmware/myproject.h')) as f, open(
-            f'{model.config.get_output_dir()}/src/firmware/{project_name}.h', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/firmware/myproject.h')) as f,
+            open(f'{model.config.get_output_dir()}/src/firmware/{project_name}.h', 'w') as fout,
+        ):
             model_inputs = model.get_input_variables()
             model_outputs = model.get_output_variables()
             # model_brams = [var for var in model.get_weight_variables() if var.storage.lower() == 'bram']
@@ -254,9 +256,10 @@ def write_defines(self, model):
             model (ModelGraph): the hls4ml model.
         """
         filedir = os.path.dirname(os.path.abspath(__file__))
-        with open(os.path.join(filedir, '../templates/oneapi/firmware/defines.h')) as f, open(
-            f'{model.config.get_output_dir()}/src/firmware/defines.h', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/firmware/defines.h')) as f,
+            open(f'{model.config.get_output_dir()}/src/firmware/defines.h', 'w') as fout,
+        ):
             for line in f.readlines():
                 # Insert numbers
                 if '// hls-fpga-machine-learning insert numbers' in line:
@@ -298,9 +301,10 @@ def write_parameters(self, model):
             model (ModelGraph): the hls4ml model.
         """
         filedir = os.path.dirname(os.path.abspath(__file__))
-        with open(os.path.join(filedir, '../templates/oneapi/firmware/parameters.h')) as f, open(
-            f'{model.config.get_output_dir()}/src/firmware/parameters.h', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/firmware/parameters.h')) as f,
+            open(f'{model.config.get_output_dir()}/src/firmware/parameters.h', 'w') as fout,
+        ):
             for line in f.readlines():
                 if '// hls-fpga-machine-learning insert includes' in line:
                     newline = line
@@ -376,9 +380,10 @@ def write_test_bench(self, model):
                     output_predictions, f'{model.config.get_output_dir()}/tb_data/tb_output_predictions.dat'
                 )
 
-        with open(os.path.join(filedir, '../templates/oneapi/myproject_test.cpp')) as f, open(
-            f'{model.config.get_output_dir()}/src/{project_name}_test.cpp', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/myproject_test.cpp')) as f,
+            open(f'{model.config.get_output_dir()}/src/{project_name}_test.cpp', 'w') as fout,
+        ):
             for line in f.readlines():
                 indent = ' ' * (len(line) - len(line.lstrip(' ')))
 
@@ -434,9 +439,10 @@ def write_bridge(self, model):
         indent = '    '
 
         filedir = os.path.dirname(os.path.abspath(__file__))
-        with open(os.path.join(filedir, '../templates/oneapi/myproject_bridge.cpp')) as f, open(
-            f'{model.config.get_output_dir()}/src/{project_name}_bridge.cpp', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/myproject_bridge.cpp')) as f,
+            open(f'{model.config.get_output_dir()}/src/{project_name}_bridge.cpp', 'w') as fout,
+        ):
             for line in f.readlines():
                 if 'MYPROJECT' in line:
                     newline = line.replace('MYPROJECT', format(project_name.upper()))
@@ -511,9 +517,10 @@ def write_build_script(self, model):
         # Makefile
         filedir = os.path.dirname(os.path.abspath(__file__))
         device = model.config.get_config_value('Part')
-        with open(os.path.join(filedir, '../templates/oneapi/CMakeLists.txt')) as f, open(
-            f'{model.config.get_output_dir()}/CMakeLists.txt', 'w'
-        ) as fout:
+        with (
+            open(os.path.join(filedir, '../templates/oneapi/CMakeLists.txt')) as f,
+            open(f'{model.config.get_output_dir()}/CMakeLists.txt', 'w') as fout,
+        ):
             for line in f.readlines():
                 line = line.replace('myproject', model.config.get_project_name())
                 line = line.replace('mystamp', model.config.get_config_value('Stamp'))