From 5b2a026beab02583211615ccbb1b0d6569d25e51 Mon Sep 17 00:00:00 2001 From: arnaudon Date: Thu, 30 Nov 2023 11:44:25 +0100 Subject: [PATCH] lint --- MARBLE/main.py | 9 ++++++--- MARBLE/plotting.py | 3 ++- MARBLE/preprocessing.py | 4 ++-- 3 files changed, 10 insertions(+), 6 deletions(-) diff --git a/MARBLE/main.py b/MARBLE/main.py index c3d38705..98bc149e 100644 --- a/MARBLE/main.py +++ b/MARBLE/main.py @@ -4,16 +4,19 @@ import warnings from datetime import datetime from pathlib import Path -import yaml -from tqdm import tqdm import torch import torch.nn.functional as F import torch.optim as opt +import yaml from torch import nn from torch_geometric.nn import MLP +from tqdm import tqdm -from MARBLE import dataloader, geometry, layers, utils +from MARBLE import dataloader +from MARBLE import geometry +from MARBLE import layers +from MARBLE import utils class net(nn.Module): diff --git a/MARBLE/plotting.py b/MARBLE/plotting.py index 099fe09b..0b1f464d 100644 --- a/MARBLE/plotting.py +++ b/MARBLE/plotting.py @@ -1,12 +1,13 @@ """Plotting module.""" import os from pathlib import Path -import torch + import matplotlib import matplotlib.pyplot as plt import networkx as nx import numpy as np import seaborn as sns +import torch from matplotlib import gridspec from matplotlib.colors import LinearSegmentedColormap from matplotlib.patches import FancyArrowPatch diff --git a/MARBLE/preprocessing.py b/MARBLE/preprocessing.py index 5ea3fa86..1fbd5060 100644 --- a/MARBLE/preprocessing.py +++ b/MARBLE/preprocessing.py @@ -125,7 +125,7 @@ def _compute_geometric_objects( gauges (nxdxd): local gauges at all points par (dict): updated dictionary of parameters local_gauges: whether to use local gauges - + """ n, dim_emb = data.pos.shape dim_signal = data.x.shape[1] @@ -184,4 +184,4 @@ def _compute_geometric_objects( ] data.L, data.Lc, data.gauges, data.local_gauges = L, Lc, gauges, local_gauges - return data \ No newline at end of file + return data