Slight bug fix in grid_search method

pyrates-neuroscience · Jul 10, 2019 · d138255 · d138255
1 parent 5558ac2
commit d138255
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyrates/utility/grid_search.py b/pyrates/utility/grid_search.py
@@ -1086,7 +1086,10 @@ def adapt_circuit(circuit: CircuitIR, params: dict, param_map: dict) -> CircuitI
                         circuit.nodes[node]['node'].op_graph.nodes[op]['variables'][var_name]['value'] = float(val)
                 else:
                     for op in circuit.nodes[node]['node'].op_graph.nodes:
-                        op['variables'][var]['value'] = float(val)
+                        try:
+                            circuit.nodes[node]['node'].op_graph.nodes[op]['variables'][var]['value'] = float(val)
+                        except KeyError:
+                            pass
 
             # change variable values on edges
             edges = param_map[key]['edges'] if 'edges' in param_map[key] else []