gather expert data

2024-09-09 17:35:21 +08:00 · 2024-09-09 17:35:21 +08:00 · 43f905e634
commit 43f905e634
parent c6c4411beb
1 changed files with 2 additions and 1 deletions
--- a/behaviors/custom/Dribble/Dribble.py
+++ b/behaviors/custom/Dribble/Dribble.py
@ -208,6 +208,7 @@ class Dribble():
            if self.phase >= WIND_DOWN_STEPS - 5:
                self.phase = 0
                return True
+        if self.obs != None and self.act != None:
            np.array(self.gather_expert_data.append((self.obs, self.act)), dtype=object)
        size = len(self.gather_expert_data)
        if size > 5e4: