cvmatrix 2.0.1__tar.gz → 2.0.2__tar.gz

{cvmatrix-2.0.1 → cvmatrix-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: cvmatrix
-Version: 2.0.1
+Version: 2.0.2
 Summary: Fast computation of possibly weighted and possibly centered/scaled training set kernel matrices in a cross-validation setting.
 License: Apache-2.0
 Author: Sm00thix

cvmatrix-2.0.2/cvmatrix/__init__.py

@@ -0,0 +1 @@
+__version__ = "2.0.2"

{cvmatrix-2.0.1 → cvmatrix-2.0.2}/cvmatrix/cvmatrix.py

@@ -731,14 +731,14 @@ class CVMatrix:
         ) and self.Y_total is not None:
             self.sum_Y_total = np.sum(self.Yw_total, axis=0, keepdims=True)
         if self.scale_X:
-            self.sum_sq_X_total = np.expand_dims(
-                np.einsum("ij, ij -> j", self.Xw_total, self.X_total), axis=0
+            self.sum_sq_X_total = np.sum(
+                self.Xw_total * self.X_total, axis=0, keepdims=True
             )
         else:
             self.sum_sq_X_total = None
         if self.scale_Y and self.Y_total is not None:
-            self.sum_sq_Y_total = np.expand_dims(
-                np.einsum("ij, ij -> j", self.Yw_total, self.Y_total), axis=0
+            self.sum_sq_Y_total = np.sum(
+                self.Yw_total * self.Y_total, axis=0, keepdims=True
             )
         else:
             self.sum_sq_Y_total = None

{cvmatrix-2.0.1 → cvmatrix-2.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cvmatrix"
-version = "2.0.1"
+version = "2.0.2"
 description = "Fast computation of possibly weighted and possibly centered/scaled training set kernel matrices in a cross-validation setting."
 authors = ["Sm00thix <oleemail@icloud.com>"]
 maintainers = ["Sm00thix <oleemail@icloud.com>"]

cvmatrix-2.0.1/cvmatrix/__init__.py

@@ -1 +0,0 @@
-__version__ = "2.0.1"