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): ...@@ -296,7 +296,7 @@ class StudentTEnergy(EnergyOperator):
def apply(self, x): def apply(self, x):
self._check_input(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: if not x.want_metric:
return res return res
met = ScalingOperator(self.domain, (self._theta+1) / (self._theta+3)) met = ScalingOperator(self.domain, (self._theta+1) / (self._theta+3))
Markdown is supported
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