add laleMaximum constraints to cca

IBM · May 4, 2021 · 5430b79 · 5430b79
1 parent 080b167
commit 5430b79
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/lale/lib/sklearn/cca.py b/lale/lib/sklearn/cca.py
@@ -14,13 +14,22 @@
             "additionalProperties": False,
             "properties": {
                 "n_components": {
-                    "type": "integer",
-                    "minimum": 1,
-                    "minimumForOptimizer": 2,
-                    "maximumForOptimizer": 256,
-                    "distribution": "uniform",
                     "default": 2,
                     "description": "number of components to keep.",
+                    "allOf": [
+                        {
+                            "type": "integer",
+                            "minimum": 1,
+                            "minimumForOptimizer": 2,
+                            "maximumForOptimizer": 256,
+                            "distribution": "uniform",
+                            "laleMaximum": "X/items/maxItems",  # number of columns (n_features)
+                        },
+                        {
+                            "type": "integer",
+                            "laleMaximum": "X/maxItems",  # number of rows (n_samples)
+                        },
+                    ],
                 },
                 "scale": {
                     "type": "boolean",