Merge pull request #170 from iivalchev/main

fix calculating E[x2] in FullSupportBarDistribution

src/tabpfn/model/bar_distribution.py

@@ -620,7 +620,7 @@ def mean_of_square(self, logits: torch.Tensor) -> torch.Tensor:
         )
         bucket_mean_of_square[-1] = (
             side_normals[1].variance
-            + (side_normals[1].variance + self.borders[-2]).square()
+            + (side_normals[1].mean + self.borders[-2]).square()
         )
         p = torch.softmax(logits, -1)
         return p @ bucket_mean_of_square