Skip to content

Commit d390298

Browse files
committed
fixed some pd saving/loading data problems with headers
1 parent 70b9bf8 commit d390298

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mimic/model_simulate/base_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def save_data(self, filename, data=None):
186186
return
187187

188188
try:
189-
pd.DataFrame(data).to_csv(filename, index=False)
189+
pd.DataFrame(data).to_csv(filename, index=False, header=False)
190190
except Exception as e:
191191
print(f"Error saving data to {filename}: {e}")
192192
return False
@@ -208,7 +208,7 @@ def load_data(self, filename):
208208
raise FileNotFoundError(f"No file found at {filename}")
209209

210210
try:
211-
self.data = pd.read_csv(filename)
211+
self.data = pd.read_csv(filename, header=None).values.tolist()
212212
except Exception as e:
213213
print(f"Error reading data from {filename}: {e}")
214214
return False

0 commit comments

Comments
 (0)