From 94bd94d4bed61042b4fba3270936b3ee2e889a55 Mon Sep 17 00:00:00 2001
From: Marek Makowski <marek@iiasa.ac.at>
Date: Tue, 23 Apr 2024 16:40:38 +0200
Subject: [PATCH] mcma: minor correction of Pllots::plot3D

to correctly handle bi-criteria cases
---
 models/mcma/plots.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/models/mcma/plots.py b/models/mcma/plots.py
index 2b56e39..663dc3f 100644
--- a/models/mcma/plots.py
+++ b/models/mcma/plots.py
@@ -296,7 +296,9 @@ def kde_stages(self):  # for each stage: histogram + KDE
         self.figures['kde_stages'] = fig
 
     def plot3D(self):
-        if self.n_crit != 3:
+        if self.n_crit < 3:      # just return for bi-criteria problem
+            return
+        if self.n_crit > 3:
             # todo: implement 3D subplots for more than 3 criteria
             print(f'Plots.plot3D(): not implemented for {self.n_crit} criteria yet.')
             return