diff --git a/.idea/csv-editor.xml b/.idea/csv-editor.xml
index 51d69c3..4f87a7d 100644
--- a/.idea/csv-editor.xml
+++ b/.idea/csv-editor.xml
@@ -24,6 +24,13 @@
+
+
+
+
+
+
+
diff --git a/.idea/deployment.xml b/.idea/deployment.xml
index 9af79fb..65480d4 100644
--- a/.idea/deployment.xml
+++ b/.idea/deployment.xml
@@ -1,6 +1,6 @@
-
+
diff --git a/environment.py b/environment.py
index 9d01b52..3cf2d45 100644
--- a/environment.py
+++ b/environment.py
@@ -159,7 +159,7 @@ class ESSEnv(gym.Env):
temperature_df = pd.read_csv('data/temper.csv', sep=',')
wind_df = pd.read_csv('data/wind.csv', sep=',')
- price = price_df['Price'].to_numpy(dtype=float)
+ price = price_df['price'].to_numpy(dtype=float)
load = load_df['houseload'].to_numpy(dtype=float)
irradiance = irradiance_df['irradiance'].to_numpy(dtype=float)
temperature = temperature_df['t2m'].to_numpy(dtype=float)
diff --git a/test.py b/test.py
index de201b3..79c7124 100644
--- a/test.py
+++ b/test.py
@@ -19,7 +19,7 @@
# model = YourModelClass() # 实例化你的模型
# model.load_state_dict(checkpoint)
-print(model)
+# print(model)
import pickle