Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
bfb135a
1
Parent(s):
9335949
Add constraints test with custom operator
Browse files- test/test.py +3 -0
test/test.py
CHANGED
@@ -50,6 +50,9 @@ class TestPipeline(unittest.TestCase):
|
|
50 |
verbosity=0,
|
51 |
**self.default_test_kwargs,
|
52 |
procs=0,
|
|
|
|
|
|
|
53 |
)
|
54 |
model.fit(self.X, y)
|
55 |
equations = model.equations
|
|
|
50 |
verbosity=0,
|
51 |
**self.default_test_kwargs,
|
52 |
procs=0,
|
53 |
+
# Test custom operators with constraints:
|
54 |
+
nested_constraints={"square_op": {"square_op": 3}},
|
55 |
+
constraints={"square_op": 10},
|
56 |
)
|
57 |
model.fit(self.X, y)
|
58 |
equations = model.equations
|