Commit a7643e6f authored by Reimar Leike's avatar Reimar Leike
Browse files

Fix a bug where studentt would only work if theta is a scalar

parent c5c2881e
Pipeline #74130 passed with stages
in 29 minutes and 53 seconds
......@@ -296,7 +296,7 @@ class StudentTEnergy(EnergyOperator):
def apply(self, x):
self._check_input(x)
res = ((self._theta+1)/2)*(x**2/self._theta).ptw("log1p").sum()
res = (((self._theta+1)/2)*(x**2/self._theta).ptw("log1p")).sum()
if not x.want_metric:
return res
met = ScalingOperator(self.domain, (self._theta+1) / (self._theta+3))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment