Merge pull request #39 from NREL/gb/better_random_seed

fixed random seed setting to be compatible with all tf 2 versions
NREL · Oct 13, 2022 · 7352fe2 · 7352fe2
2 parents 2e06723 + 916c4df
commit 7352fe2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/phygnn/base.py b/phygnn/base.py
@@ -3,6 +3,7 @@
 Custom Neural Network Infrastructure.
 """
 from abc import ABC, abstractmethod
+import random
 import os
 import pickle
 import pprint
@@ -241,9 +242,9 @@ def seed(s=0):
         s : int
             Random seed
         """
+        random.seed(s)
         np.random.seed(s)
         tf.random.set_seed(s)
-        tf.keras.utils.set_random_seed(s)
 
     @classmethod
     def get_val_split(cls, *args, shuffle=True, validation_split=0.2):

diff --git a/phygnn/model_interfaces/base_model.py b/phygnn/model_interfaces/base_model.py
@@ -3,6 +3,7 @@
 Base Model Interface
 """
 from abc import ABC
+import random
 import copy
 import pprint
 import logging
@@ -446,10 +447,10 @@ def seed(s=0):
         s : int
             Random number generator seed
         """
+        random.seed(s)
         np.random.seed(s)
         if TF2:
             tf.random.set_seed(s)
-            tf.keras.utils.set_random_seed(s)
         else:
             tf.random.set_random_seed(s)
 

diff --git a/phygnn/version.py b/phygnn/version.py
@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Physics Guided Neural Network version."""
 
-__version__ = '0.0.21'
+__version__ = '0.0.22'