Skip to content

Commit 88f04f6

Browse files
authored
Merge pull request #2 from feslima/dev
add feature:Auxiliary expressions.(Author: F lima)
2 parents 31d6e56 + e2e07e2 commit 88f04f6

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

gui/models/data_storage.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ class DataStorage(QObject):
8686
}
8787
_EXPR_ALIAS_TYPES = {
8888
'cv': 'Candidate (CV)',
89+
'aux': 'Auxiliary',
8990
'cst': 'Constraint function',
9091
'obj': 'Objective function (J)'
9192
}
@@ -1491,7 +1492,8 @@ def check_simulation_setup(self):
14911492
# get expressions validity
14921493
expr_df = self.expression_table_data
14931494
expr_valid_check = expr_df['Expression'].apply(
1494-
lambda x: is_expression_valid(x, aliases['Alias'].tolist())
1495+
lambda x: is_expression_valid(x, aliases['Alias'].tolist() + expr_df['Alias'].tolist()
1496+
)
14951497
)
14961498

14971499
is_name_not_duplicated = pd.concat(
@@ -1644,6 +1646,10 @@ def evaluate_expr_data(self, sampled_data: pd.DataFrame,
16441646
expr_row_values[expr['Alias']] = expr_to_parse.evaluate(
16451647
row_val_dict)
16461648

1649+
row_val_dict.update(
1650+
{expr['Alias']: expr_row_values[expr['Alias']]}
1651+
)
1652+
16471653
# append values to expr_df
16481654
expr_df = expr_df.append(expr_row_values, ignore_index=True)
16491655

gui/models/sampling.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -442,6 +442,7 @@ def model_function(self, x):
442442
expr_to_parse = parser.parse(expr['Expression'])
443443
var_list = expr_to_parse.variables()
444444
expr_values[expr['Alias']] = expr_to_parse.evaluate(results)
445+
results.update(expr_values)
445446

446447
# separate constraints values
447448
con_aliases = expr_data.loc[

0 commit comments

Comments
 (0)