Skip to content

Instantly share code, notes, and snippets.

#define EMPTY 0
#define NORMAL_FOOD 1
#define INTERACTIVE_FOOD 2
// Anything above Wall should be obstacles
#define WALL 3
#define AGENTS 4
#define LOG_BUFFER_SIZE 8192
#define SET_BIT(arr, i) (arr[(i) / 8] |= (1 << ((i) % 8)))
MAKE_FNS = {
'breakout': lambda: lazy_import('pufferlib.ocean.breakout.breakout', 'Breakout'),
'pong': lambda: lazy_import('pufferlib.ocean.pong.pong', 'Pong'),
'enduro': lambda: lazy_import('pufferlib.ocean.enduro.enduro', 'Enduro'),
'moba': lambda: lazy_import('pufferlib.ocean.moba.moba', 'Moba'),
'nmmo3': lambda: lazy_import('pufferlib.ocean.nmmo3.nmmo3', 'NMMO3'),
'snake': lambda: lazy_import('pufferlib.ocean.snake.snake', 'Snake'),
'squared': lambda: lazy_import('pufferlib.ocean.squared.squared', 'Squared'),
'pysquared': lambda: lazy_import('pufferlib.ocean.squared.pysquared', 'PySquared'),
'connect4': lambda: lazy_import('pufferlib.ocean.connect4.connect4', 'Connect4'),