def placeholder_from_space()

in safe_rl/pg/network.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def placeholder_from_space(space):
    if isinstance(space, Box):
        return placeholder(space.shape)
    elif isinstance(space, Discrete):
        return tf.placeholder(dtype=tf.int32, shape=(None,))
    raise NotImplementedError('bad space {}'.format(space))