added 'check_forward_dict'

Author: abdelabd

hls4ml/converters/__init__.py

@@ -270,11 +270,14 @@ def convert_from_pytorch_model(model, input_shape, output_dir='my-hls-test', pro
 
     return pytorch_to_hls(config)
 
+def check_forward_dict(model, forward_dictionary):
+    for key in forward_dictionary:
+        assert(hasattr(model, key))
 def convert_from_pyg_model(model, n_node, node_dim, n_edge, edge_dim,
                            forward_dictionary=None, activate_final=None,
                            output_dir='my-hls-test', project_name='myproject',
                            fpga_part='xcku115-flvb2104-2-i', clock_period=5, io_type='io_parallel', hls_config={}):
-    
+    check_forward_dict(model, forward_dictionary)
     config = create_vivado_config(
         output_dir=output_dir,
         project_name=project_name,

hls4ml/converters/pyg/__init__.py

@@ -116,14 +116,12 @@ def pyg_to_hls(config):
     aggr_count = 0
     forward_dict_new = OrderedDict()
     for key, val in forward_dict.items():
-        if val=="NodeBlock":
+        if val == "NodeBlock":
             aggr_count += 1
             aggr_key = f"aggr{aggr_count}"
             aggr_val = "Aggregate"
             forward_dict_new[aggr_key] = aggr_val
         forward_dict_new[key] = val
-    print(f"forward_dict: {forward_dict}")
-    print(f"forward_dict_new: {forward_dict_new}")
 
     # complete the layer list
     for i, (key, val) in enumerate(forward_dict_new.items()):