Skip to content

Commit 5754837

Browse files
authored
Merge pull request p-christ#43 from shuferhoo/master
fix space.Discrete type error
2 parents 8a0dd27 + 292b653 commit 5754837

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utilities/Parallel_Experience_Generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class Parallel_Experience_Generator(object):
1616
def __init__(self, environment, policy, seed, hyperparameters, action_size, use_GPU=False, action_choice_output_columns=None):
1717
self.use_GPU = use_GPU
1818
self.environment = environment
19-
self.action_types = "DISCRETE" if self.environment.action_space.dtype == int else "CONTINUOUS"
19+
self.action_types = "DISCRETE" if self.environment.action_space.dtype in [int, 'int64'] else "CONTINUOUS"
2020
self.action_size = action_size
2121
self.policy = policy
2222
self.action_choice_output_columns = action_choice_output_columns

0 commit comments

Comments
 (0)