fix lint

agosztolai · Nov 30, 2023 · 3dedb01 · 3dedb01
1 parent 1891ba0
commit 3dedb01
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/MARBLE/geometry.py b/MARBLE/geometry.py
@@ -465,7 +465,7 @@ def compute_laplacian(data, normalization="rw"):
 
 
 def compute_connection_laplacian(data, R, normalization="rw"):
-    """Connection Laplacian
+    r"""Connection Laplacian
 
     Args:
         data: Pytorch geometric data object.
@@ -520,7 +520,7 @@ def compute_connection_laplacian(data, R, normalization="rw"):
 
 
 def compute_gauges(data, dim_man=None, n_geodesic_nb=10, n_workers=1):
-    """Orthonormal gauges for the tangent space at each node, and connection
+    r"""Orthonormal gauges for the tangent space at each node, and connection
     matrices between each pair of adjacent nodes.
 
     R is a block matrix, where the row index is the gauge we want to align to,
@@ -573,7 +573,7 @@ def _compute_gauges(inputs, i):
 
 
 def compute_connections(data, gauges, n_workers=1):
-    """Find smallest rotations R between gauges pairs. It is assumed that the first
+    r"""Find smallest rotations R between gauges pairs. It is assumed that the first
     row of edge_index is what we want to align to, i.e.,
     gauges(i) = gauges(j)@R[i,j].T
 

diff --git a/MARBLE/plotting.py b/MARBLE/plotting.py
@@ -1,7 +1,4 @@
 """Plotting module."""
-import os
-from pathlib import Path
-
 import matplotlib
 import matplotlib.pyplot as plt
 import networkx as nx