Seed the random number generator

khorshuheng · khorshuheng · commit ea397daeea4a · 2020-10-22T16:39:23.000+08:00
Signed-off-by: Khor Shu Heng &lt;khor.heng@gojek.com&gt;
diff --git a/tests/e2e/test_historical_features.py b/tests/e2e/test_historical_features.py
@@ -14,6 +14,8 @@
 from feast.data_format import ParquetFormat
 from feast.staging.storage_client import get_staging_client
 
+np.random.seed(0)
+
 
 @pytest.fixture(scope="function")
 def staging_path(pytestconfig, tmp_path):
@@ -61,8 +63,8 @@ def test_historical_features(feast_client: Client, staging_path: str):
     creation_date = retrieval_date - timedelta(1)
 
     customers = [1001, 1002, 1003, 1004, 1005]
-    daily_transactions = [np.random.rand(seed=0) * 10 for _ in customers]
-    total_transactions = [np.random.rand(seed=0) * 100 for _ in customers]
+    daily_transactions = [np.random.rand() * 10 for _ in customers]
+    total_transactions = [np.random.rand() * 100 for _ in customers]
 
     transactions_df = pd.DataFrame(
         {
