diff --git a/examples/mnist-pytorch/client/train.py b/examples/mnist-pytorch/client/train.py index d8297de0d..9ac9cce61 100644 --- a/examples/mnist-pytorch/client/train.py +++ b/examples/mnist-pytorch/client/train.py @@ -5,10 +5,9 @@ import torch from model import load_parameters, save_parameters +from data import load_data from fedn.utils.helpers.helpers import save_metadata -from .data import load_data - dir_path = os.path.dirname(os.path.realpath(__file__)) sys.path.append(os.path.abspath(dir_path)) diff --git a/examples/mnist-pytorch/client/validate.py b/examples/mnist-pytorch/client/validate.py index e2fb097eb..09328181f 100644 --- a/examples/mnist-pytorch/client/validate.py +++ b/examples/mnist-pytorch/client/validate.py @@ -4,10 +4,9 @@ import torch from model import load_parameters +from data import load_data from fedn.utils.helpers.helpers import save_metrics -from .data import load_data - dir_path = os.path.dirname(os.path.realpath(__file__)) sys.path.append(os.path.abspath(dir_path)) diff --git a/pyproject.toml b/pyproject.toml index 7eb4d0a11..7e3484a8f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -178,6 +178,7 @@ lint.ignore = [ "B007", # Loop control variable `v` not used within loop body "N806", # Variable `X_test` in function should be lowercase "SIM103", # Return the condition directly + "I001", # [*] Import block is un-sorted or un-formatted # unsafe? "S104", # Possible binding to all interfaces