Skip to content
Snippets Groups Projects
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
No related branches found
No related tags found
1 merge request!455Fix a bug where studentt would only work if theta is a scalar
Pipeline #74130 passed
......@@ -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))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment