rename for pypi

Author: MMathisLab

AROS/__init__.py aros_node/__init__.pyAROS/__init__.py renamed to aros_node/__init__.py

@@ -1,13 +1,13 @@
 
-import aros
+import aros_node
 import argparse
 import torch
 import torch.nn as nn
-from aros.evaluate import *
-from aros.utils import *
+from aros_node.evaluate import *
+from aros_node.utils import *
 from tqdm.notebook import tqdm
-from aros.data_loader import *
-from aros.stability_loss_function import *
+from aros_node.data_loader import *
+from aros_node.stability_loss_function import *
 
 def main():
     parser = argparse.ArgumentParser(description="Hyperparameters for the script")

AROS/data_loader.py aros_node/data_loader.pyAROS/data_loader.py renamed to aros_node/data_loader.py

@@ -3,7 +3,7 @@
 from torch.utils.data import DataLoader, Subset
 from torchvision.datasets import CIFAR10, CIFAR100
 from torchvision.transforms import ToTensor
-from aros import (
+from aros_node import (
     LabelChangedDataset,
     get_subsampled_subset,
     get_loaders,