From bbcff1ec183ffbb028ef063b0cd8d2124c8d056b Mon Sep 17 00:00:00 2001
From: Giorgio Morales Luna <giorgiomoralesluna@gmail.com>
Date: Thu, 21 Mar 2024 14:37:23 -0600
Subject: [PATCH] Formatted as a Python package

---
 DemoPaper.ipynb                                      | 4 ++--
 PIGenerator.py                                       | 2 +-
 src/PredictionIntervals/Datasets/GenerateDatasets.py | 2 +-
 src/PredictionIntervals/Trainer/TrainNN.py           | 2 +-
 src/PredictionIntervals/models/NNModel.py            | 2 +-
 src/{ => PredictionIntervals}/utils.py               | 0
 6 files changed, 6 insertions(+), 6 deletions(-)
 rename src/{ => PredictionIntervals}/utils.py (100%)

diff --git a/DemoPaper.ipynb b/DemoPaper.ipynb
index aca07b8..15eb410 100644
--- a/DemoPaper.ipynb
+++ b/DemoPaper.ipynb
@@ -38,14 +38,14 @@
     }
    ],
    "source": [
-    "import src.utils\n",
+    "import src.PredictionIntervals.utils as utils\n",
     "import pandas as pd\n",
     "from scipy import stats\n",
     "from PIGenerator import PIGenerator\n",
     "import matplotlib.pyplot as plt\n",
     "name = 'Synth'    # Define the name of the dataset \n",
     "crossval = '5x2'  # We'll use 5x2 cross-validation for all methods \n",
-    "_, _, _, _ = src.utils.create_synth_data(plot=True)  # Plot the dataset"
+    "_, _, _, _ = utils.create_synth_data(plot=True)  # Plot the dataset"
    ]
   },
   {
diff --git a/PIGenerator.py b/PIGenerator.py
index 3ce30b8..5d04f68 100644
--- a/PIGenerator.py
+++ b/PIGenerator.py
@@ -1,7 +1,7 @@
 import os
 import sys
 import time
-from src import utils
+from src.PredictionIntervals import utils
 import torch
 import pickle
 import itertools
diff --git a/src/PredictionIntervals/Datasets/GenerateDatasets.py b/src/PredictionIntervals/Datasets/GenerateDatasets.py
index bbf280c..735a7ca 100644
--- a/src/PredictionIntervals/Datasets/GenerateDatasets.py
+++ b/src/PredictionIntervals/Datasets/GenerateDatasets.py
@@ -1,6 +1,6 @@
 import numpy as np
 import pandas as pd
-from src.utils import create_synth_data
+from src.PredictionIntervals.utils import create_synth_data
 
 
 class DataLoader:
diff --git a/src/PredictionIntervals/Trainer/TrainNN.py b/src/PredictionIntervals/Trainer/TrainNN.py
index f69b6fa..468874a 100644
--- a/src/PredictionIntervals/Trainer/TrainNN.py
+++ b/src/PredictionIntervals/Trainer/TrainNN.py
@@ -1,6 +1,6 @@
 import torch
 import pickle
-from src.utils import *
+from src.PredictionIntervals.utils import *
 from sklearn.model_selection import KFold
 from src.PredictionIntervals.models.NNModel import NNModel
 
diff --git a/src/PredictionIntervals/models/NNModel.py b/src/PredictionIntervals/models/NNModel.py
index a8c2484..cc6da8c 100644
--- a/src/PredictionIntervals/models/NNModel.py
+++ b/src/PredictionIntervals/models/NNModel.py
@@ -4,7 +4,7 @@
 import pickle
 import random
 import numpy as np
-from src import utils
+from src.PredictionIntervals import utils
 from tqdm import trange
 from torch import optim
 from src.PredictionIntervals.models.network import *
diff --git a/src/utils.py b/src/PredictionIntervals/utils.py
similarity index 100%
rename from src/utils.py
rename to src/PredictionIntervals/utils.py