File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -2029,7 +2029,7 @@ def _create_seed(self, user_specified_seed):
20292029 if user_specified_seed is not None :
20302030 return user_specified_seed
20312031 elif getattr (_SEED_GENERATOR , "generator" , None ):
2032- return _SEED_GENERATOR .generator .randint (1 , 1e9 )
2032+ return _SEED_GENERATOR .generator .randint (1 , int ( 1e9 ) )
20332033 else :
20342034 return random .randint (1 , int (1e9 ))
20352035
Original file line number Diff line number Diff line change @@ -78,9 +78,9 @@ def get_random_seed():
7878 the random seed as an integer.
7979 """
8080 if getattr (backend ._SEED_GENERATOR , "generator" , None ):
81- return backend ._SEED_GENERATOR .generator .randint (1 , 1e9 )
81+ return backend ._SEED_GENERATOR .generator .randint (1 , int ( 1e9 ) )
8282 else :
83- return random .randint (1 , 1e9 )
83+ return random .randint (1 , int ( 1e9 ) )
8484
8585
8686def is_tensor_or_tensor_list (v ):
You can’t perform that action at this time.
0 commit comments