fix(builder.py): fix seed setting (#32)

PKU-Alignment · Mar 26, 2023 · 0e720b3 · 0e720b3
1 parent 7d78e4c
commit 0e720b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/safety_gymnasium/builder.py b/safety_gymnasium/builder.py
@@ -152,7 +152,7 @@ def _get_task(self) -> BaseTask:
 
     def set_seed(self, seed: int = None) -> None:
         """Set internal random state seeds."""
-        self._seed = np.random.randint(2**32) if seed is None else seed
+        self._seed = np.random.randint(2**32, dtype='int64') if seed is None else seed
         self.task.random_generator.set_random_seed(self._seed)
 
     def reset(