Merge pull request #138 from adamhall/main

Changed instances of randint to integers as per issue #137
utiasDSL · Oct 12, 2023 · d286cea · d286cea
2 parents 74e9056 + 7266c26
commit d286cea
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/safe_control_gym/controllers/mpc/gp_mpc.py b/safe_control_gym/controllers/mpc/gp_mpc.py
@@ -712,7 +712,7 @@ def learn(self,
                                                  self.train_iterations + validation_iterations,
                                                  random_state=self.seed)
             input_samples = np.array(input_samples)  # not being used currently
-            seeds = self.env.np_random.randint(0, 99999, size=self.train_iterations + validation_iterations)
+            seeds = self.env.np_random.integers(0, 99999, size=self.train_iterations + validation_iterations)
             for i in range(self.train_iterations + validation_iterations):
                 # For random initial state training.
                 #init_state = init_state_samples[i,:]

diff --git a/safe_control_gym/envs/disturbances.py b/safe_control_gym/envs/disturbances.py
@@ -100,7 +100,7 @@ def reset(self,
               env
               ):
         if self.step_offset is None:
-            self.current_step_offset = self.np_random.randint(self.max_step)
+            self.current_step_offset = self.np_random.integers(self.max_step)
         else:
             self.current_step_offset = self.step_offset
         self.current_peak_step = int(self.current_step_offset + self.duration / 2)
@@ -146,7 +146,7 @@ def reset(self,
               env
               ):
         if self.step_offset is None:
-            self.current_step_offset = self.np_random.randint(self.max_step)
+            self.current_step_offset = self.np_random.integers(self.max_step)
         else:
             self.current_step_offset = self.step_offset