Update import in nn/utils.py

tractometry · Dec 17, 2024 · 0aa5180 · 0aa5180
1 parent 6ea8283
commit 0aa5180
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/afqinsight/nn/utils.py b/afqinsight/nn/utils.py
@@ -2,11 +2,12 @@
 import tensorflow as tf
 import torch
 import torch.nn as nn
-from neurocombat_sklearn import CombatModel
 from sklearn.impute import SimpleImputer
 from sklearn.model_selection import train_test_split
 from tensorflow.keras import layers
 
+from afqinsight.neurocombat_sklearn import CombatModel
+
 
 def extract_layer_info_pytorch(layer):
     info = {}