fix implicit imports

spock/additional_feature_functions.py

@@ -1,8 +1,9 @@
 from collections import OrderedDict
 
 import numpy as np
-from AMD_functions import AMD, AMD_crit
-from feature_functions import find_strongest_MMR, get_pairs, populate_trio
+
+from .AMD_functions import AMD, AMD_crit
+from .feature_functions import find_strongest_MMR, get_pairs, populate_trio
 
 
 def additional_get_tseries(sim, args):

spock/spock_reg_model.py

@@ -1,6 +1,8 @@
 import json
 import math
 import pickle as pkl
+import warnings
+from collections import OrderedDict
 from copy import deepcopy as copy
 
 import numpy as np
@@ -10,17 +12,13 @@
 from sklearn.preprocessing import PowerTransformer, StandardScaler
 from torch import nn
 from torch.autograd import Variable
+from torch.optim.optimizer import Optimizer
 
 try:
     from torch._six import inf
 except ModuleNotFoundError:
     from torch import inf
 
-import warnings
-from collections import OrderedDict
-
-from torch.optim.optimizer import Optimizer
-
 
 class CustomOneCycleLR(torch.optim.lr_scheduler._LRScheduler):
     """Custom version of one-cycle learning rate to stop early"""