Fix stat recorder checks

RyanNavillus · Apr 28, 2024 · 08c9e0b · 08c9e0b
1 parent 8c4eeb7
commit 08c9e0b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/syllabus/core/curriculum_base.py b/syllabus/core/curriculum_base.py
@@ -29,8 +29,7 @@ def __init__(self, task_space: TaskSpace, random_start_tasks: int = 0, task_name
         self.completed_tasks = 0
         self.task_names = task_names
         self.n_updates = 0
-        if record_stats:
-            self.stat_recorder = StatRecorder(self.task_space)
+        self.stat_recorder = StatRecorder(self.task_space) if record_stats else None
 
         if self.num_tasks == 0:
             warnings.warn("Task space is empty. This will cause errors during sampling if no tasks are added.")