shankerram3 commited on
Commit
8c6fae6
·
verified ·
1 Parent(s): 2f38b12

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. server/wildfire_environment.py +2 -1
server/wildfire_environment.py CHANGED
@@ -396,7 +396,8 @@ class WildfireEnvironment(Environment):
396
  st = self._state
397
  burning = self._burning_count()
398
  burned = sum(1 for v in st.grid if v == 0)
399
- return WildfireObservation(
 
400
  grid=st.grid[:],
401
  width=self.w,
402
  height=self.h,
 
396
  st = self._state
397
  burning = self._burning_count()
398
  burned = sum(1 for v in st.grid if v == 0)
399
+ # Use model_construct to bypass Pydantic validation for dataclass/Pydantic compatibility
400
+ return WildfireObservation.model_construct(
401
  grid=st.grid[:],
402
  width=self.w,
403
  height=self.h,