seed_everything

Lightning-Sandbox · Feb 26, 2024 · 7e54e12 · 7e54e12
1 parent b8c76e3
commit 7e54e12
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/torch/model/test_estimators.py b/test/torch/model/test_estimators.py
@@ -19,6 +19,7 @@
 import pytest
 import pandas as pd
 import numpy as np
+from lightning import seed_everything
 
 from gluonts.dataset.repository import get_dataset
 from gluonts.model.predictor import Predictor
@@ -309,6 +310,7 @@ def test_estimator_constant_dataset(
     ],
 )
 def test_estimator_with_features(estimator_constructor):
+    seed_everything(42)
     freq = "1h"
     prediction_length = 12