prashantmatlani commited on
Commit
f5165d9
·
1 Parent(s): 7c8585e

fix: handle step before reset (state_data None)

Browse files
Files changed (1) hide show
  1. app/env.py +5 -0
app/env.py CHANGED
@@ -76,6 +76,11 @@ class CustomerSupportEnv:
76
 
77
 
78
  def step(self, action: dict):
 
 
 
 
 
79
 
80
  reward = 0.0
81
  done = False
 
76
 
77
 
78
  def step(self, action: dict):
79
+
80
+ # ✅ SAFETY: ensure environment initialized
81
+ if self.state_data is None:
82
+ print("⚠️ step() called before reset — auto-resetting", flush=True)
83
+ self.reset()
84
 
85
  reward = 0.0
86
  done = False