def make_session()

in maddpg/common/tf_util.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def make_session(num_cpu):
    """Returns a session that will use <num_cpu> CPU's only"""
    tf_config = tf.ConfigProto(
        inter_op_parallelism_threads=num_cpu,
        intra_op_parallelism_threads=num_cpu)
    return tf.Session(config=tf_config)