Fix bug

wesselb · Oct 8, 2021 · f484f2c · f484f2c
1 parent c1a551f
commit f484f2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mlkernels/kernels/ceq.py b/mlkernels/kernels/ceq.py
@@ -21,6 +21,9 @@ def __init__(self, alpha):
     def _compute(self, dists):
         return (1 - B.erf(self.alpha * dists / 4)) * B.exp(-0.5 * dists ** 2)
 
+    def render(self, formatter):
+        return f"CEQ({formatter(self.alpha)})"
+
     @property
     def _stationary(self):
         return True

diff --git a/tests/kernels/test_ceq.py b/tests/kernels/test_ceq.py
@@ -9,7 +9,7 @@ def test_eq():
     # Verify that the kernel has the right properties.
     assert k.alpha == 1.0
     assert k.stationary
-    assert str(k) == "CEQ()"
+    assert str(k) == "CEQ(1.0)"
 
     # Test equality.
     assert CEQ(1.0) == CEQ(1.0)