From 9f821713e8b10bbd461b21b261cf839a13c266f0 Mon Sep 17 00:00:00 2001
From: CaradryanLiang <caradryan2022@gmail.com>
Date: Fri, 17 May 2024 16:28:15 -0700
Subject: [PATCH] fix fatal bugs

---
 diffusers/scripts/train_gsa.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/diffusers/scripts/train_gsa.py b/diffusers/scripts/train_gsa.py
index 31126c4..cf6b208 100644
--- a/diffusers/scripts/train_gsa.py
+++ b/diffusers/scripts/train_gsa.py
@@ -81,6 +81,7 @@ def preprocess(member, non_member):
     x = np.vstack((member, non_member))
     y = np.concatenate((member_y_np, nonmember_y_np))
     x = preprocessing.scale(x)
+    x = np.nan_to_num(x, nan=0)
     return x, y
 
 def train_xgboost(member_features, nonmember_features):