diff --git a/pydynamo/core/system.py b/pydynamo/core/system.py index fa7629dbfd572cd623a4f1362e1a557977356ae9..984fb20ed3300b5228628709723f599b761506bc 100644 --- a/pydynamo/core/system.py +++ b/pydynamo/core/system.py @@ -121,7 +121,9 @@ class System: f = getattr(self, 'tabhl_' + name) x = np.linspace(f.xl, f.xh, 100) ylabel = name + '\n'+ self.get_comment(name) - xlabel = argname + '\n'+ self.get_comment(argname) + argop = self.eqs['update'][name]['args']['fun']['tabhl']['val'].strip() + argop = re.sub('^\(|\)$', '',re.sub('\.[jk]', '', argop)).strip() + xlabel = argop + ('\n'+ self.get_comment(argname) if argop==argname else '') return x, f(x), ylabel, xlabel, f'tabhl_{name}' def iter_all_eqs(self):