Merge pull request #14 from grok-ai/feature/tests

Return the seed resolved from the seed_index

Author: Flegyas

src/nn_core/common/utils.py

@@ -95,13 +95,15 @@ def enforce_tags(tags: Optional[List[str]]) -> List[str]:
     return tags
 
 
-def seed_index_everything(train_cfg: DictConfig, sampling_seed: int = 42) -> None:
+def seed_index_everything(train_cfg: DictConfig, sampling_seed: int = 42) -> Optional[int]:
     if "seed_index" in train_cfg and train_cfg.seed_index is not None:
         seed_index = train_cfg.seed_index
         np.random.seed(sampling_seed)
         seeds = np.random.randint(np.iinfo(np.int32).max, size=max(42, seed_index + 1))
         seed = seeds[seed_index]
         seed_everything(seed)
         pylogger.info(f"Setting seed {seed} from seeds[{seed_index}]")
+        return seed
     else:
         pylogger.warning("The seed has not been set! The reproducibility is not guaranteed.")
+        return None