fix pyright

Farama-Foundation · Jun 28, 2023 · 1661e87 · 1661e87
1 parent e3677f1
commit 1661e87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/minari/dataset/minari_dataset.py b/minari/dataset/minari_dataset.py
@@ -192,7 +192,7 @@ def __init__(
         self._additional_data_id = 0
         if episode_indices is None:
             episode_indices = np.arange(self._data.total_episodes)
-        self.episode_indices = episode_indices
+        self.episode_indices: np.ndarray = episode_indices
 
         self.spec = MinariDatasetSpec(
             env_spec=self._data.env_spec,