folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix : check_components_type (#227)

Browse Source 
fixing components variable in plotter.py
This commit is contained in:
M.Couraud
2024-02-06 10:39:50 +01:00
committed by GitHub
parent 20b2ed1890
commit 1464eee891
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pina/plotter.py
View File

@@ -158,8 +158,9 @@ class Plotter:
Plot sample of SolverInterface output. Plot sample of SolverInterface output.
:param SolverInterface solver: The ``SolverInterface`` object instance. :param SolverInterface solver: The ``SolverInterface`` object instance.
:param list(str) components: The output variable to plot. If None, all :param str | list(str) components: The output variable(s) to plot.
the output variables of the problem are selected. Default value is None. If None, all the output variables of the problem are selected.
Default value is None.
:param dict fixed_variables: A dictionary with all the variables that :param dict fixed_variables: A dictionary with all the variables that
should be kept fixed during the plot. The keys of the dictionary should be kept fixed during the plot. The keys of the dictionary
are the variables name whereas the values are the corresponding are the variables name whereas the values are the corresponding
@@ -176,6 +177,13 @@ class Plotter:
if components is None: if components is None:
components = solver.problem.output_variables components = solver.problem.output_variables
if isinstance(components, str):
components = [components]
if not isinstance(components, list):
raise NotImplementedError('Output variables must be passed'
'as a string or a list of strings.')
if len(components) > 1: if len(components) > 1:
raise NotImplementedError('Multidimensional plots are not implemented, ' raise NotImplementedError('Multidimensional plots are not implemented, '
'set components to an available components of the problem.') 'set components to an available components of the problem.')