fix imports

Author: MMathisLab

aros_node/evaluate.py

@@ -1,7 +1,7 @@
 import numpy as np
 from tqdm.notebook import tqdm
 from sklearn.metrics import roc_curve, roc_auc_score, precision_recall_curve, auc
-from utils import *
+from aros_node.utils import *
 import argparse
 import torch
 import torch.nn as nn

aros_node/stability_loss_function.py

@@ -2,8 +2,8 @@
 from robustbench.utils import load_model
 import torch.nn as nn
 from torch.nn.parameter import Parameter
-import utils
-from utils import *
+import aros_node.utils
+from aros_node.utils import *
 from torch.utils.data import DataLoader, Dataset, TensorDataset, Subset, SubsetRandomSampler, ConcatDataset
 import numpy as np
 from tqdm.notebook import tqdm

setup.cfg

@@ -1,5 +1,5 @@
 [metadata]
-name = aros
+name = aros-node
 version = 0.0.1
 author = Hossein Mirzaei, Mackenzie Mathis
 author_email = [email protected]