def __init__()

in custom_envs/obstructedmaze_gamut.py [0:0]

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

	def __init__(self, distribution='easy', max_difficulty=None, seed=1337):

		self.distribution = distribution
		if distribution == 'easy':
			self.max_difficulty = 3
		elif distribution == 'medium':
			self.max_difficulty = 6
		elif distribution == 'hard':
			self.max_difficulty = 9
		else:
			raise ValueError(f'Unsupported distribution {distribution}.')

		if max_difficulty is not None:
			self.max_difficulty = max_difficulty

		self.subenvs = []
		for env_name in ALL_SUBENVS[:self.max_difficulty]:
			self.subenvs.append(gym.make(env_name))

		self.num_subenvs = len(self.subenvs)

		self.seed(seed)
		self.reset()