From 0aa5180f92cdae2783dc787bff7d5d757e78355d Mon Sep 17 00:00:00 2001
From: Howard Chiu <137316255+chiuhoward@users.noreply.github.com>
Date: Tue, 17 Dec 2024 13:43:35 +0800
Subject: [PATCH] Update import in nn/utils.py

---
 afqinsight/nn/utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/afqinsight/nn/utils.py b/afqinsight/nn/utils.py
index 1692f34..34ffb6f 100644
--- 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 = {}