clean up

LarsKue · Dec 26, 2023 · bab5d98 · bab5d98
1 parent 0d1b642
commit bab5d98
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 11 deletions.
diff --git a/src/losses/mmd.py b/src/losses/mmd.py
@@ -46,7 +46,7 @@ def mmd_loss(samples: Tensor, target: Tensor, kernel=gaussian_kernel, scales: Te
 
     Returns
     -------
-    mmd: Maximum-Mean-Discrepancy between data and code. Tensor of shape ()
+    mmd: Maximum-Mean-Discrepancy between data and code. Tensor of shape (N,)
     """
     if scales == "auto":
         scales = torch.logspace(-6, 6, 13)

diff --git a/src/visualization/__init__.py b/src/visualization/__init__.py
@@ -1,5 +1,4 @@
 
-
 from .multiscatter import multiscatter, multiscatter_bp
 from .rainbow import Rainbow
 from .scatter import scatter, scatter_bp
diff --git a/src/visualization/multiscatter.py b/src/visualization/multiscatter.py
@@ -43,4 +43,4 @@ def multiscatter_bp(samples: np.ndarray, layout: (int, int), **render_kwargs):
 
     plt.subplots_adjust(left=0.0, right=1.0, bottom=0.0, top=1.0, wspace=0.0, hspace=0.0)
 
-    return fig
+    return fig
diff --git a/src/visualization/rainbow.py b/src/visualization/rainbow.py
@@ -38,11 +38,3 @@ def __call__(self, X, alpha=None, bytes=False):
         rgba = np.concatenate([rgb, alpha], axis=1)
 
         return rgba
-
-
-r, g, b = 1, 1, 1
-
-m = max(r, g, b)
-
-
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -43,4 +43,4 @@ def multiscatter_bp(samples: np.ndarray, layout: (int, int), **render_kwargs):

		plt.subplots_adjust(left=0.0, right=1.0, bottom=0.0, top=1.0, wspace=0.0, hspace=0.0)

		return fig
		return fig